推荐看原创
原文链接
在安装完NVIDIA驱动后,紧接着就是安装cuda和cudnn。其实安装的过程还是很简单的,下载合适的版本一步一步来进行安装即可。
安装cuda
- 到官网下载适合自己的cuda版本。官网链接为:cuda官网链接
下图是我选择的版本。
- 安装cuda
通过如下命令进行安装:
sudo chmod a+x cuda_9.0.176_384.81_linux.run
sudo sh cuda_9.0.176_384.81_linux.run
由于安装cuda是在我的游戏本上安装的,在安装过程中兵没有将过程的图片截取下来,再加上写博客是在我的Mac上写的,所以借用我在安装过程中参考的一篇博客的图,在此表示感谢!虽然这个图的cuda版本和我安装的不同,但是本质还是一样的,同样可以参考。
- 添加到PATH
在参考的博客中,博主加入的是~/.bashrc,但为了能够使得所有用户都是用,我直接配置到/etc/profile文件中。在/etc/profile文件末尾添加如下内容
export LD_LIBRARY_PATH=$LDLIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
保存退出,source /etc/profile
即可生效。
通过nvcc --version即可显示CUDA的版本信息。
如果显示,则安装成功。
安装cudnn
- 下载cudnn
同样需要根据自己安装的cuda选择适配的cudnn版本。官网链接为cuDNN下载链接
由于官网下载需要登陆,没有账号的铁子们可以先进行注册,登陆进去即可获取看到下载列表,选择合适的版本进行下载即可。我选择了7.6.0版。
- 安装cudnn
下载下来的为一个压缩包,直接对该压缩包进行解压即可:
tar -zxvf cudnn-9.0-linux-x64-v7.6.0.64.tgz
cudnn的安装就是拷贝cudnn的一些文件到cuda:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
一个简单的拷贝过程就完成了cudnn的安装。这样cudnn就安装完成了。