-
首先nvidia-smi看一下有没有nvidai驱动。没有就装一个。
-
如果正常,则有驱动。这时也会出现cuda版本(右上角),忽略它;
-
nvcc -V 查看是否有cuda(nvidia-smi的cuda和nvcc的cuda不是一个,见参考)
-
安装cuda,在这里找到需要的cuda版本;
-
利用 run.sh 格式进行安装,这种安装可以只安装cuda tool,而不安装nvidia驱动此说明源自这里;
-
按照此教程安装tool
-
由于已存在驱动,则会报错,继续按照,注意别安装驱动(他前面别打×)
-
成功后会出现 Summary,其中Driver为未选择
-
设置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}}
-
安装成功:nvcc -V 检查
-
下载cudnn地址,注意要注册账号,注意下载的版本要对应,记得下载cuDNN Library for Linux;
-
注意,这里无法用wget下载,因为要用账号。参考此文章进行下载,只需要把对应的版本号改一下,版本号在步骤11处可以得到。
-
按照教程解压、复制即可(注意复制到上面安装的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*
-
利用检测cuda/cudnn评论区方法查看安装结果:
cat /usr/local/cuda-11.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
ubuntu cuda/cudnn 最新版安装及检查方法(亲测有效,2021-06-25)
猜你喜欢
转载自blog.csdn.net/qq_40491305/article/details/118216669
今日推荐
周排行