Ubuntu 16 server 安装 tensorflow-GPU

时间:2022-08-01 17:20:41

目录

本次使用主机配置:

cpui7-8700K     显卡GTX-1080Ti

一、安装显卡驱动

打开命令窗口(ctrl+Alt+T)

sudo apt-get purge nvidia*

sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update sudo apt-get install nvidia-384 nvidia-settings

如果报错add-apt-repository不存在,运行以下代码解决:

sudo apt-get install software-properties-common python-software-properties

检验是否安装成功

nvidia-smi

显示以下结果表明安装成功

Ubuntu 16 server 安装 tensorflow-GPU

二、安装CUDA

由于目前NVIDIA官网上的CUDA9.0+版本 存在缺陷,本次安装选择CUDA8.0

下载安装包:

地址:https://developer.nvidia.com/cuda-80-ga2-download-archive

Ubuntu 16 server 安装 tensorflow-GPU

Ubuntu 16 server 安装 tensorflow-GPU

进入下载页面后,下载CUDA8.0(cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb)安装包和

一个升级包(cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb)

安装步骤

1、安装基础包

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb sudo apt-get update sudo apt-get install cuda 

2、安装升级包

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb sudo apt-get update sudo apt-get upgrade cuda

设置环境变量:

在.bashrc中添加

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} export CUDA_HOME=/usr/local/cuda

查看安装版本

cat /usr/local/cuda/version.txt

输出

CUDA Version 8.0.61

三、安装cuDNN

cuDNN6.0和CUDA8.0最佳拍档,所以本次选择cuDNN6.0

下载安装包

地址:https://developer.nvidia.com/rdp/cudnn-download

Ubuntu 16 server 安装 tensorflow-GPU

安装:

1、将下载的压缩包复制到CUDA目录

sudo cp cudnn-8.0-linux-x64-v6.0.tgz /usr/local/cuda

2、进入/usr/local/cuda解压文件,然后建立连接

tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d

查看cuDNN版本信息:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

输出

#define CUDNN_MAJOR      6
#define CUDNN_MINOR      0
#define CUDNN_PATCHLEVEL 21
--
#define CUDNN_VERSION    (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL) #include "driver_types.h"

检测CUDA和cuDNN是否安装成功:

进入测试目录

//进入测试目录:
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery //编译环境
make -j4 //运行:
./deviceQuery

结果中有Result = PASS 表示安装成功

Ubuntu 16 server 安装 tensorflow-GPU

四、安装tensorflow-gpu

//安装命令,要安装哪个版本的tensorflow-gpu,使用 ‘tensorflow-gpu==xx’就行了
pip install tensorflow-gpu==1.4

//卸载命令
pip uninstall tensorflow-gpu

测试gpu是否安装成功

import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

输出里有 GPU的名称,显存等信息,表示tensorflow可以使用GPU了

Ubuntu 16 server 安装 tensorflow-GPU

 到此安装已经完成,如有问题,欢迎留言讨论。