ubuntu16.04+GTX1080TI+CUDA9.0+CUDNN7.0
@ 环境:hp,ubuntu16.04 ,GTX1080TI
一、安装NVIDIA显卡
打开命令窗口(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
安装完毕后重启电脑,运行nvidia-smi,查看生效的显卡驱动
二、安装CUDA
1、官网下载cuda9.0(cuda_9.0.176_384.81_linux.run):下载链接
2、cd 到cuda下载目录进行安装。
sudo sh cuda_9.0.176_384.81_linux.run
(1)按q键跳过安装须知,输入accept接受。
(2)问你驱动装不装,输入n,选择不装,上面装过了。
(3)问你CUDA装不装,输入y,选择安装。回车键安装在默认位置(也可以自己选择位置安装)。
(4)问你要不要创建链接,输入y,选择创建。
(5)问你sample装不装,因为我们只是调用CUDA,不去写CUDA程序,输入n,选择不安装。
3、添加路径库:
(1)打开bashrc:
sudo vim ~/.bashrc
(2)添加路径:
export PATH="/usr/local/cuda-9.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH"
(3)使之生效:
source ~/.bashrc
三、安装CUDNN7.0
1、官网下载CUDNN7.0(cudnn-9.0-linux-x64-v7.4.1.5.tgz):下载链接
2、下载cudnn7.0的linux包(tgz格式)。
3、cd 到下载路径解压文件:
tar -xzvf cudnn-9.0-linux-x64-v7.4.1.5.tgz
4、复制文件:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
5、添加权限:
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
到此,全部安装结束,接下来就可以安装tensorflow,caffe等深度学习框架了。
四、卸载及删除安装
1、cudnn:直接remove掉就可以了。
2、cuda:sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl
3、nvidia驱动:nvidia-uninstall
五、查看版本
1、CUDA :
cat /usr/local/cuda/version.txt
2、 CUDNN :
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
参考:
1、https://blog.csdn.net/weixin_40294256/article/details/79157838