Ubuntu安装cuda与cudnn
时间: 2023-09-30 13:05:34 浏览: 60
安装 CUDA 和 cuDNN 是进行深度学习的必备步骤。下面是在 Ubuntu 上安装 CUDA 和 cuDNN 的步骤:
### 安装 CUDA
1. 下载 CUDA:访问官网 `https://developer.nvidia.com/cuda-downloads`,选择您的操作系统、体系结构和 CUDA 版本,下载 CUDA。
2. 安装 CUDA:运行以下命令进行安装:
```bash
sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb
sudo apt-get update
sudo apt-get install cuda
```
3. 设置环境变量:将以下内容添加到 `~/.bashrc` 文件中:
```bash
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
4. 重新加载 `.bashrc` 文件:运行以下命令重新加载 `.bashrc` 文件:
```bash
source ~/.bashrc
```
5. 验证 CUDA 安装:运行以下命令以确认 CUDA 是否正确安装:
```bash
nvcc -V
```
如果您看到 CUDA 版本信息,则说明 CUDA 安装成功。
### 安装 cuDNN
1. 下载 cuDNN:访问官网 `https://developer.nvidia.com/rdp/cudnn-download`,选择您的操作系统、体系结构和 cuDNN 版本,下载 cuDNN。
2. 解压缩 cuDNN:运行以下命令解压缩 cuDNN:
```bash
tar -xzvf cudnn-<version>.tgz
```
3. 复制文件:运行以下命令将文件复制到 CUDA 的安装目录中:
```bash
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 /usr/local/cuda/lib64/libcudnn*
```
注意,如果您的 CUDA 安装目录不是 `/usr/local/cuda`,则需要相应地更改路径。
4. 验证 cuDNN 安装:运行以下命令以确认 cuDNN 是否正确安装:
```bash
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
```
如果您看到 cuDNN 版本信息,则说明 cuDNN 安装成功。
希望这些步骤对您有所帮助。