linux ubuntu系统下安装CUDA cuDNN

原创已于 2023-05-20 07:22:10 修改 · 2.4k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu #linux #运维

于 2022-09-16 15:26:39 首次发布

本文详细指导了如何下载、安装CUDA及cuDNN，并说明了如何通过软链接实现不同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. 完毕

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理