ubantu 安装cudnn
时间: 2023-09-26 22:13:18 浏览: 52
要在Ubuntu上安装cuDNN,您可以按照以下步骤进行操作:
1. 首先,确保您已经安装了NVIDIA显卡驱动程序。可以使用以下命令来安装:
```
sudo apt-get install nvidia-driver-<version>
```
其中`<version>`是您所需的驱动程序版本号。
2. 访问NVIDIA开发者网站(https://developer.nvidia.com/cudnn)并下载适用于您的CUDA版本的cuDNN软件包。请确保选择与您的CUDA版本兼容的cuDNN版本。
3. 下载完成后,解压缩cuDNN压缩包。可以使用以下命令:
```
tar -xzvf cudnn-<version>.tgz
```
其中`<version>`是您下载的cuDNN版本号。
4. 进入解压缩后的cuDNN目录,复制cuDNN文件到CUDA的安装路径下:
```
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
```
5. 接下来,添加cuDNN的库路径到LD_LIBRARY_PATH环境变量。可以使用以下命令:
```
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
```
6. 现在,cuDNN已成功安装在您的Ubuntu系统上。您可以通过编译运行使用cuDNN的CUDA程序来验证安装。
请注意,以上步骤假设您已经在系统上正确安装了CUDA。如果您还没有安装CUDA,请先按照适当的方式安装CUDA,并确保与您下载的cuDNN版本兼容。