wsl2安装cudnn
时间: 2023-08-24 14:07:08 浏览: 236
tensorflow1.3以后的gpu安装cudnn
1. 下载CUDNN
首先需要到NVIDIA官网下载对应版本的CUDNN。下载地址为:https://developer.nvidia.com/rdp/cudnn-download。
需要注意的是,CUDNN的版本需要和CUDA的版本对应。比如,如果安装的是CUDA 11.2,那么需要下载对应的CUDNN版本。
2. 解压CUDNN
下载完成后,将压缩文件解压到任意目录。
例如,将CUDNN解压到/opt/cudnn目录下:
```
sudo tar -xzvf cudnn-11.2-linux-x64-v8.2.2.26.tgz -C /opt/cudnn/
```
3. 配置环境变量
打开终端,输入以下命令:
```
sudo gedit ~/.bashrc
```
在打开的文件中添加以下内容:
```
export LD_LIBRARY_PATH=/opt/cudnn/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/opt/cudnn/cuda/bin:$PATH
```
保存并关闭文件,然后执行以下命令使配置生效:
```
source ~/.bashrc
```
4. 验证CUDNN是否安装成功
打开终端,输入以下命令:
```
cat /opt/cudnn/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
```
如果输出结果中含有以下内容,则说明CUDNN安装成功:
```
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 2
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
```
至此,CUDNN安装完成。
阅读全文