安装及配置过程
一、下载安装CUDA Toolkit
- 1.查看操作系统版本及支持CUDA版本
- 2.官网下载并安装对应版本CUDA
- 3.配置环境变量
- 4.测试CUDA安装是否成功
二、下载安装cuDNN
- 1.官网下载对应版本cuDNN
一、下载安装CUDA Toolkit
1.查看操作系统版本及支持CUDA版本
1)查看系统版本
uname -a
2)查看系统支持CUDA版本
图中标红处说明此系统支持CUDA最高版本为:11.0,即下载CUDA时版本要控制在11.0以下
nvidia-smi
2.官网下载并安装对应版本CUDA
1)根据系统支持版本下载对应版本的CUDA Toolkit,作者此处选择CUDA10.2。官网链接
2)选择所需版本,通过对应命令进行下载安装(注意此处需要记住下载文件的目录,之后需要找到)
按照顺序输入上图 下方的两条指令
在输入第二条命令之后,经过短暂的等待,会出现用户安装界面,
其中包括是否选择安装Nvidia显卡驱动,如果本地已有驱动,可选择不安装(将光标移至Driver,点击Enter即可),
之后移至Install处,点击Enter即可进行安装。
3.配置环境变量
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即可成功切换。
- 测试CUDA安装是否成功
出现如下字样即说明安装成功
nvcc -V
二、下载安装cuDNN
1.官网下载对应版本cuDNN
1)对应上述安装的CUDA版本,选择合适的cuDNN版本,官网链接
- 此处需要注册Nvidia账户
- 要将下方的版本号修改成自己的版本
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