安装CUDA 11.0和cuDNN 8.0步骤
时间: 2024-01-14 19:04:17 浏览: 353
tensorflowGpu-2.3.0-cuda11.0-cudnn8.0-vs2019.rar
以下是在Ubuntu 20.04系统上安装CUDA 11.0和cuDNN 8.0的步骤:
1. 下载CUDA 11.0
您可以从NVIDIA官网下载CUDA 11.0安装包。选择适合您的系统的安装包,并下载。
2. 安装CUDA 11.0
双击安装包进行安装。按照安装向导的指示进行操作。在安装过程中,您可以选择安装CUDA Toolkit、CUDA Samples等组件。
3. 配置环境变量
安装完成后,您需要将CUDA路径添加到系统环境变量中。打开终端,执行以下命令:
```
sudo vim ~/.bashrc
```
在打开的文件中添加以下两行代码:
```
export PATH=/usr/local/cuda-11.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH
```
保存并退出文件。执行以下命令使环境变量生效:
```
source ~/.bashrc
```
4. 下载cuDNN 8.0
您可以从NVIDIA官网下载cuDNN 8.0的压缩包。选择与您的CUDA版本和操作系统相匹配的版本,并下载。
5. 安装cuDNN 8.0
解压缩cuDNN 8.0的压缩包,并将其中的文件复制到CUDA的安装目录中。打开终端,执行以下命令:
```
cd ~/Downloads
tar -xzvf cudnn-11.0-linux-x64-v8.0.4.30.tgz
sudo cp -P cuda/include/cudnn*.h /usr/local/cuda-11.0/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64
sudo chmod a+r /usr/local/cuda-11.0/include/cudnn*.h /usr/local/cuda-11.0/lib64/libcudnn*
```
安装完成后,您可以继续安装TensorFlow-GPU,并使用GPU加速训练模型。
阅读全文