ubuntu16.04安装gtx1080ti驱动+配置CUDA8
1.将所需要的文件下载下来
下载gtx1080ti驱动
进入nvidia的官网http://www.nvidia.cn/Download/Find.aspx?lang=cn,找到GTX080Ti的驱动下载:
下载CUDA8.0
从官网下载CUDA8.0https://developer.nvidia.com/cuda-80-ga2-download-archive
需要下载两个文件,一个是它的base一个是Patch分支文件
安装注意先后顺序,先安装base后安装patch
下载CUDDNv6
CUDDNv6下载地址:https://developer.nvidia.com/rdp/cudnn-download
先注册帐号,然后才能下载
二、安装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))
若不报错,则成功安装