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

猜你喜欢

转载自blog.csdn.net/qq_40491305/article/details/118216669