ubantu配置cuDNN
时间: 2024-08-14 19:09:14 浏览: 102
ubuntu安装显卡驱动+cuda+cudnn教程(18.04+cuda10+cudnn9.05为例)
5星 · 资源好评率100%
Ubuntu上配置CUDA和cuDNN通常是为了支持深度学习在GPU上的加速。以下是基本步骤:
1. **安装CUDA**:
- 首先,访问NVIDIA的CUDA官网下载对应版本的CUDA Toolkit,适用于Ubuntu的Debian或Ubuntu二进制包。
- 安装时需要遵循官方文档,一般涉及到添加NVIDIA GPG密钥、添加APT源,然后通过`sudo apt-get install cuda`命令安装。
2. **更新主机文件**:
更新主机文件,以便系统能够识别GPU设备,打开`/etc/hosts`文件并添加类似 `06:00.0 NVIDIA GPU` 的显卡地址。
3. **安装cuDNN**:
- CUDA Toolkit中会包含cuDNN库的基本依赖,但如果需要最新版本,你需要从NVIDIA的cuDNN官网下载cuDNN SDK,并解压到CUDA的lib64(或对应CUDA版本路径)下。
- 按照cuDNN的安装指南配置环境变量。
4. **验证安装**:
- 运行`nvcc --version`检查是否能看到CUDA版本,以及`cudnnVersion()`(如果在Python中使用TensorFlow或PyTorch)确认cuDNN版本。
5. **设置环境变量**:
- 将CUDA和cuDNN的路径添加到系统的PATH环境变量中,例如编辑`~/.bashrc`或`~/.bash_profile`。
6. **重启终端**:
为了应用新配置,关闭当前终端窗口,然后打开一个新的终端。
阅读全文