1. 下载cuda https://developer.nvidia.com/cuda-toolkit-archive
2. 安装cuda
3. 安装完毕之后,配置环境变量
sudo vim ~/.bashrc
4. 末尾添加
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
export PATH="/usr/local/cuda/bin:$PATH"
export CUDA_HOME="/usr/local/cuda:$CUDA_HOME"
5.注意,上面路径中是用/usr/local/cuda
而不是/cuda-11.3
,因为接下来需要通过软链接,以实现多个CUDA版本共存。输入下面代码,即可完成软链接的生成,其中/cuda-11.3
替换成自己的cuda安装目录名称。
a. 删除之前生成的软链接
sudo rm -rf /usr/local/cuda
b. 生成新的软链接
sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda
之后如果需要更换CUDA版本,只需再执行以上两条命令,替换软链接就行。
6. cuDNN安装 下载 https://developer.nvidia.com/rdp/cudnn-archive
7.下载解压出来
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
8. 完毕