cudnn8.1在ubuntu安装
时间: 2024-10-17 18:02:00 浏览: 26
适用于tensorflow-2.11.0 CUDA版本11.2的cuDNN8.1版本
CUDNN (Compute Unified Device Architecture Neural Network) 是一种用于GPU加速深度学习神经网络计算的库。要在Ubuntu上安装CUDA 8.1及其对应的cudnn 8.1版本,你需要按照以下步骤操作:
1. **更新包列表并安装依赖**:
```bash
sudo apt-get update
sudo apt-get install build-essential cuda-command-line-tools-8-0
```
2. **下载CUDA 8.1**:
访问NVIDIA官网(https://developer.nvidia.com/cuda-downloads),选择适合Ubuntu 16.04 LTS的CUDA 8.1版本下载DEB文件。
3. **安装CUDA**:
使用`dpkg`命令安装下载的.deb文件:
```bash
sudo dpkg -i nvidia*.deb
```
4. **配置环境变量**:
需要编辑`~/.bashrc`或`.profile`文件添加cuda路径到系统PATH中:
```bash
echo "export PATH=/usr/local/cuda-8.1/bin:$PATH" >> ~/.bashrc
source ~/.bashrc
```
5. **安装cuDNN**:
CUDA Toolkit通常包含了一个名为`cudnn-8.1-linux-x64-v5.1.tgz`的cuDNN文件。下载并解压,然后运行`sudo ./install_cudnn.sh`进行安装。这一步可能会提示你输入cuDNN的license key。
6. **确认安装**:
查看cuDNN是否安装成功,可以在终端中输入`nvidia-smi`或`cudnnVersion`检查。
阅读全文