ubuntu cuda/cudnn 最新版安装及检查方法(亲测有效,2021-06-25)

别裁伪体亲风雅,转益多师是汝师。这篇文章主要讲述ubuntu cuda/cudnn 最新版安装及检查方法(亲测有效,2021-06-25)相关的知识,希望能为你提供帮助。

  1. 首先nvidia-smi看一下有没有nvidai驱动。没有就装一个。
  2. 如果正常,则有驱动。这时也会出现cuda版本(右上角),忽略它;
  3. nvcc -V 查看是否有cuda(nvidia-smi的cuda和nvcc的cuda不是一个,见参考)
  4. 安装cuda,在这里找到需要的cuda版本;
  5. 利用 run.sh 格式进行安装,这种安装可以只安装cuda tool,而不安装nvidia驱动此说明源自这里;
  6. 按照此教程安装tool
  7. 由于已存在驱动,则会报错,继续按照,注意别安装驱动(他前面别打×)
  8. 成功后会出现 Summary,其中Driver为未选择
  9. 设置path路径(summary中有安装的路径,按照其修改下面代码)
    export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64{LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
  10. 安装成功:nvcc -V 检查
  11. 下载cudnn地址,注意要注册账号,注意下载的版本要对应,记得下载cuDNN Library for Linux;
  12. 注意,这里无法用wget下载,因为要用账号。参考此文章进行下载,只需要把对应的版本号改一下,版本号在步骤11处可以得到。
  13. 按照教程解压、复制即可(注意复制到上面安装的11.1路径),注意新版指令发生了变化,见此评论区sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.1/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.1/lib64 sudo chmod a+r /usr/local/cuda-11.1/include/cudnn*.h /usr/local/cuda-11.1/lib64/libcudnn*
  14. 利用检测cuda/cudnn评论区方法查看安装结果:
    cat /usr/local/cuda-11.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
【ubuntu cuda/cudnn 最新版安装及检查方法(亲测有效,2021-06-25)】

    推荐阅读