ununtu20.04安装cudnn
时间: 2024-11-09 12:13:01 浏览: 7
Ubuntu 20.04 CUDA&cuDNN安装方法(图文教程)
5星 · 资源好评率100%
在Ubuntu 20.04上安装CUDA和cuDNN通常需要几个步骤,因为它们都是用于支持深度学习在GPU上的运算库。以下是安装过程的一个简要概述:
1. **更新系统**:
```
sudo apt update
sudo apt upgrade
```
2. **安装CUDA**:
- 首先访问NVIDIA官网下载适用于Ubuntu 20.04的CUDA版本,然后按照指示进行下载并安装。
- 添加NVIDIA官方PPA (个人软件包存档):
```bash
sudo add-apt-repository ppa:graphics-drivers/ppa
```
- 更新软件源:
```
sudo apt update
```
- 安装CUDA:
```
sudo apt install cuda
```
3. **验证安装**:
```bash
nvidia-smi
```
4. **安装cuDNN**:
- 下载cuDNN对应CUDA版本的文件,从NVIDIA Developer网站获取。
- 解压并移动到合适的路径(如`/usr/local/cuda/lib64`):
```bash
sudo cp /path/to/cuDNN/*.so /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/lib64/*cuda*.so
```
5. **添加环境变量**:
- 编辑`~/.bashrc` 或 `~/.bash_profile` 并添加:
```bash
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64
```
6. **使更改生效**:
```bash
source ~/.bashrc
```
安装完成后,检查cuDNN是否成功安装,可以运行示例程序或者通过命令`nvidia-smi | grep cudnn`查看。
阅读全文