Ubuntu16.04 RTX2060安装cuda9.0+cudnn7.60

  在安装完NVIDIA驱动后,紧接着就是安装cuda和cudnn。其实安装的过程还是很简单的,下载合适的版本一步一步来进行安装即可。

安装cuda

  • 到官网下载适合自己的cuda版本。官网链接为:cuda官网链接
    下图是我选择的版本。
    Ubuntu16.04 RTX2060安装cuda9.0+cudnn7.60
  • 安装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版本和我安装的不同,但是本质还是一样的,同样可以参考。
    Ubuntu16.04 RTX2060安装cuda9.0+cudnn7.60
  • 添加到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版。
    Ubuntu16.04 RTX2060安装cuda9.0+cudnn7.60
    Ubuntu16.04 RTX2060安装cuda9.0+cudnn7.60
  • 安装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就安装完成了。