ubuntu22.04 docker安装cudnn
时间: 2023-06-28 15:07:45 浏览: 135
ubuntu安装docker
5星 · 资源好评率100%
安装CUDNN需要以下步骤:
1. 下载相应的CUDNN库文件,可以在NVIDIA的官方网站上下载。需要注意的是,下载的文件要与你的CUDA版本相对应。
2. 将下载的文件解压缩到合适的目录下,例如:/usr/local/cuda。
3. 将解压后的文件分别复制到CUDA的安装目录下。
```
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*
```
4. 配置环境变量,将以下内容添加到~/.bashrc文件中:
```
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
5. 让环境变量生效:`source ~/.bashrc`
6. 确认是否安装成功。可以通过以下命令来验证:
```
nvcc -V
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
```
如果都能显示相应的版本信息,则说明安装成功。
注意:如果在Docker中安装CUDNN,需要将CUDNN库文件复制到Docker容器中,并且在Dockerfile中添加相应的安装步骤。
阅读全文