cudnn安装ubuntu20.04
时间: 2023-11-04 15:05:25 浏览: 87
Ubuntu20环境安装, Cuda11.1 + Cudnn
5星 · 资源好评率100%
安装cudnn需要以下步骤:
1. 下载cudnn安装包,需要先注册NVIDIA开发者账号。
2. 解压安装包并复制文件到CUDA的安装目录。
3. 修改文件权限和环境变量。
具体步骤如下:
1. 访问NVIDIA官网,注册一个开发者账号。
2. 下载适用于Ubuntu 20.04的cudnn安装包,下载链接:https://developer.nvidia.com/rdp/cudnn-download。
3. 解压下载的安装包,进入解压后的目录。
4. 复制cudnn文件到CUDA的安装目录,命令如下:
```
sudo cp -P include/cudnn*.h /usr/local/cuda/include
sudo cp -P lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
```
5. 配置环境变量,命令如下:
```
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
```
6. 验证cudnn是否安装成功,命令如下:
```
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
```
如果输出以下内容,则表示安装成功:
```
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 5
```
阅读全文