ubuntu16.04安装gtx1080ti驱动+配置CUDA8.0+CUDDNv6

1.将所需要的文件下载下来

下载gtx1080ti驱动

进入nvidia的官网http://www.nvidia.cn/Download/Find.aspx?lang=cn,找到GTX080Ti的驱动下载:


ubuntu16.04安装gtx1080ti驱动+配置CUDA8.0+CUDDNv6_第1张图片

下载CUDA8.0

从官网下载CUDA8.0https://developer.nvidia.com/cuda-80-ga2-download-archive

需要下载两个文件,一个是它的base一个是Patch分支文件

安装注意先后顺序,先安装base后安装patch

ubuntu16.04安装gtx1080ti驱动+配置CUDA8.0+CUDDNv6_第2张图片

下载CUDDNv6

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

先注册帐号,然后才能下载


ubuntu16.04安装gtx1080ti驱动+配置CUDA8.0+CUDDNv6_第3张图片

二、安装gtx1080ti驱动

禁用nouveau。

sudo vim /etc/modprobe.d/blacklist.conf

在文件末尾添加如下几行:

blacklist vga16fb

blacklist nouveau

blacklist rivafb

blacklist rivatv

blacklist nvidiafb

然后sudoapt-get upgrade

更新完后重启系统,确认nouveau是已经被屏蔽掉,使用lsmod命令查看,若已经被屏蔽掉就不会有输出

lsmod| grep nouveau

然后进入命令行模式进行安装:sudo./NVIDIA-Linux-x86_64-384.59.run

三、安装CUDA8.0

进入CUDA8.0目录进行下列操作

sudo dpkg-icuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb

sudo apt-getupdate

sudo apt-getinstall cuda

sudo dpkg-icuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb

sudo apt-getupdatesudo apt-getupgrade cuda

在 ~/.bashrc 中设置环境变量:

sudovim ~/.bashrc

在最后面添加exportPATH=/usr/local/cuda/bin${PATH:+:${PATH}}exportLD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}exportCUDA_HOME=/usr/local/cuda

运行source ~/.bashrc使其生效

四、安装CUDDNv6.0

解压安装包

tar-zxvfcudnn-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

安装TensorFlow

CUDA8.0+CUDDNv6对应的TensorFlow版本是1.4,因此在安装的时候需要指定版本

pip install tensorflow-gpu==1.4

安装完成后进行测试

importt ensorflow as tfhello = tf

constant('Hello, TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

若不报错,则成功安装

你可能感兴趣的:(ubuntu16.04安装gtx1080ti驱动+配置CUDA8.0+CUDDNv6)