3_ ubuntu16.04下安装 CUDA8.0 + CDDNN6.0

本人电脑的配置详情,见我的另一篇文章: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

结果如下图:

猜你喜欢

转载自blog.csdn.net/kongli524/article/details/85055759