本人电脑的配置详情,见我的另一篇文章:2_ ubuntu下安装显卡驱动:https://blog.csdn.net/kongli524/article/details/85054893
一、安装CUDA8.0
参考网址:https://blog.csdn.net/enjoyyl/article/details/78086459#%E5%AE%89%E8%A3%85cudnn
1.下载NVIDIA CUDA Toolkit安装包
(1) 16.04以后的版本可以在这里找:https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal
出现下图,选择对应版本,一定要选runfile,类似下图:
最后下载Base Installer版本安装包,终端执行相应的:sudo sh XX.run文件。转接下面的2部分:安装文件.run执行后。
(2) 历史版本在这里找:点击上面(a)图中右下方的 Legacy Releases,出现下图:
点击,需要的版本,我选的是CUDA Toolkit 8.0 GA2,然后选择对应操作系统等,最后下载Base Installer版本安装包执行该.run文件。
我的选择如下图,点击Base Installer后的Downloads,下载完毕,进入到cuda_8.0.61_375.26_linux.run下载的目录下,终端输入:sudo sh cuda_8.0.61_375.26_linux.run 。转接下面的2部分:安装文件.run执行后。
2.安装文件.run执行后:
注意这里第二个让建立NVIDIA驱动时一定要选n,不然刚安好的驱动就用不了了,其他正常选择accept或yes,与默认地址按enter键就行。具体操作见下图:
3.添加环境变量
终端输入:sudo gedit ~/.bashrc,用来打开“./bashrc”文件,在文件末尾加上以下内容并保存:
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
添加完毕,在终端输入:source ~/.bashrc,接着重启系统。
4.检查CUDA版本
终端输入:cat /usr/local/cuda/version.txt
结果如下图:
5.查看cuda设备:
终端输入:
cd NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery
make
./deviceQuery
输出如下:
卸载cuda8.0:
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
sudo rm -r /usr/local/cuda-8.0 (删除残余)
二、安装CUDNN6.0
参考网址:https://blog.csdn.net/enjoyyl/article/details/78086459#%E5%AE%89%E8%A3%85cudnn
1.官网下载cudnn安装包
官网地址:https://developer.nvidia.com/cudnn,第一次登录需要注册。
点击 “DOWNLOAD cuDNN”,然后出现下图,选择同意,再点击“Archived cuDNN Releases”,
出现下图,选择自己需要的版本就可以了,
我选择的是“Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0”,然后出现下图选择“cuDNN v6.0 Library for Linux”,就会自动下载。
我下载的是cudnn-8.0-linux-x64-v6.0.tgz。
2.解压
进入“cudnn-8.0-linux-x64-v6.0.tgz”所在目录,解压文件,然后将其移动到你需要的位置;
3.copy文件至CUDA安装目录
终端输入:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64
至此,安装完成。
4.查看版本
cudnn 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
结果如下图: