ubuntu2004安装cuda和cudnn手把手
时间: 2023-11-10 15:02:31 浏览: 195
安装CUDA和cuDNN是在Ubuntu 20.04上进行深度学习开发的重要步骤。下面是手把手的安装步骤:
1. 安装CUDA:
- 首先,打开终端并更新系统软件包列表:
```
sudo apt-get update
```
- 在安装CUDA之前,需要确保系统上安装了合适的显卡驱动程序。可以通过以下命令检查当前安装的驱动程序版本:
```
nvidia-smi
```
- 接下来,下载CUDA Toolkit的runfile安装文件。在NVIDIA官方网站上,选择适合你系统的CUDA版本,然后下载对应的runfile安装文件。
- 安装CUDA Toolkit。打开终端,进入到下载目录,并运行以下命令(假设下载文件为cuda_installer.run):
```
chmod +x cuda_installer.run
sudo ./cuda_installer.run
```
- 运行安装程序后,会提示你接受许可协议和选择安装选项。按照提示进行操作,默认选项通常是可以接受的。
- 安装完成后,需要将CUDA的路径添加到系统环境变量中。打开终端,编辑 .bashrc 文件:
```
nano ~/.bashrc
```
- 在文件末尾添加以下行并保存:
```
export PATH=/usr/local/cuda-<version>/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-<version>/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
- 替换 `<version>` 为你所安装的CUDA版本号。
- 使环境变量生效:
```
source ~/.bashrc
```
- 验证CUDA是否正确安装。运行以下命令查看CUDA版本信息:
```
nvcc --version
```
2. 安装cuDNN:
- 首先,登录NVIDIA开发者网站并下载适合你的系统的cuDNN包(需要注册并同意许可协议)。
- 下载完成后,解压文件并进入解压后的目录。
- 复制cuDNN文件到CUDA的安装目录:
```
sudo cp include/cudnn*.h /usr/local/cuda/include
sudo cp lib64/libcudnn* /usr/local/cuda/lib64
```
- 接下来,为cuDNN创建软链接:
```
sudo ln -s /usr/local/cuda/lib64/libcudnn* /usr/local/cuda/lib64/
```
- 最后,修改cuDNN的权限:
```
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
```
3. 安装完成后,你可以重新启动系统,然后开始使用CUDA和cuDNN进行深度学习开发。
阅读全文