sudo 安装cudnn
时间: 2024-08-13 16:02:46 浏览: 162
sudo 安装 CUDA 和 cuDNN 的一般步骤如下:
1. **更新系统**:
首先确保你的Linux发行版是最新的,因为cuDNN依赖于CUDA,所以需要先安装CUDA。
2. **下载CUDA**:
访问NVIDIA官网下载适用于你系统的CUDA版本,通常可以从`https://developer.nvidia.com/cuda-downloads`下载对应CUDA SDK和Driver。
3. **安装CUDA**:
使用`sudo`运行安装脚本,例如对于Ubuntu或Debian系的系统,可能会看到类似这样的命令:
```bash
sudo sh NVIDIA-Linux-x86_64-<version>.run
```
按照提示一路跟随,可能需要接受许可协议并选择合适的组件安装。
4. **添加路径**:
安装完成后,记得添加CUDA路径到环境变量,这通常会自动完成,但也可能需要手动编辑`~/.bashrc`或`~/.bash_profile`添加相应的行。
5. **安装cuDNN**:
获取cuDNN库,通常从NVIDIA Developer网站获取。解压并移动到CUDA Toolkit的`include`和`lib64`目录下。然后,你可以通过CUDA管理工具如`nvidia-smi`或`nvcc`来确认cuDNN是否成功安装。
6. **设置环境变量**:
对于cuDNN,也需要将包含路径添加到环境变量中。这可以通过编辑`LD_LIBRARY_PATH`或其他相关的环境变量配置文件来实现。
7. **验证安装**:
最后,通过在新打开的终端窗口中运行`python -c "import torch; print(torch.cuda.is_available())"`检查是否可以正确识别cuDNN。
阅读全文