赞
踩
1)查看系统版本
uname -a
2)查看系统支持CUDA版本
图中标红处说明此系统支持CUDA最高版本为:11.0,即下载CUDA时版本要控制在11.0以下
nvidia-smi
1)根据系统支持版本下载对应版本的CUDA Toolkit,作者此处选择CUDA10.2。官网链接
2)选择所需版本,通过对应命令进行下载安装(注意此处需要记住下载文件的目录,之后需要找到)
按照顺序输入上图 下方的两条指令
在输入第二条命令之后,经过短暂的等待,会出现用户安装界面,
其中包括是否选择安装Nvidia显卡驱动,如果本地已有驱动,可选择不安装(将光标移至Driver,点击Enter即可),
之后移至Install处,点击Enter即可进行安装。
1)进入系统home文件夹,找到 .bashrc 文件,添加环境变量(其中涉及到版本问题,均与自己安装版本对应即可)
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
2)激活配置文件
source ~/.bashrc
3)如果本地安装了多个版本的CUDA,在切换版本时,可通过更改上述文件中CUDA版本即可,如将cuda-11.1改为cuda-10.1即可成功切换。
nvcc -V
1)对应上述安装的CUDA版本,选择合适的cuDNN版本,官网链接
2)选中对应版本之后,下载对应Tar包(记住你的下载路径)
3)进入Tar包下载目录,对其进行解压(解压结果类似下图)
tar -xvf cudnn-linux-x86_64-8.4.0.27_cuda10.2-archive.tar.xz
4)将cudnn安装目录下部分文件复制到cuda的安装目录下
sudo cp cudnn-linux-x86_64-8.4.0.27_cuda10.2-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-linux-x86_64-8.4.0.27_cuda10.2-archive/lib/libcudnn* /usr/local/cuda/lib64
5)修改权限
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
参考链接:https://blog.csdn.net/qq_43665602/article/details/125752433
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。