linux ubuntu系统下安装CUDA cuDNN

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. 完毕

猜你喜欢

转载自blog.csdn.net/xiangfengl/article/details/126890986