ubuntu安装 cudnn
时间: 2024-08-28 19:00:37 浏览: 97
在Ubuntu上安装cuDNN通常需要以下步骤:
1. **下载cuDNN**:首先访问NVIDIA官网(https://developer.nvidia.com/cudnn),下载适合你的CUDA版本的cuDNN。注意选择适用于Ubuntu的版本,并根据你的GPU架构(比如CPU系列)选择合适的包。
2. **更新依赖**:在安装之前,确保你的系统已经是最新的,可以使用`sudo apt update && sudo apt upgrade`命令。
3. **安装依赖**:cuDNN有时需要特定的依赖项才能安装,比如build-essential、cuda-cubin-tools等,可以使用`sudo apt install build-essential cuda-command-line-tools libncurses5-dev libcurl4-openssl-dev`。
4. **解压和配置**:将下载的cuDNN压缩包解压到合适的位置,然后创建一个名为`install.txt`的文件,内容类似`-- Silence is golden - /usr/local/cuda`,这告诉cuDNN安装在哪里。
5. **运行安装脚本**:进入解压目录,执行`bash install_file_name.run`,这里填写你刚才创建的安装文件名。
6. **完成安装**:脚本会引导你完成剩余的安装过程,包括添加环境变量和设置库路径。
7. **验证安装**:安装完成后,你可以按照前面提到的方法检查cuDNN的安装位置、环境变量设置以及运行测试程序来验证安装是否成功。
8. **添加开机启动**:为了每次启动都无需手动配置,可以将cuDNN的相关环境变量添加到`~/.bashrc`或`~/.profile`文件中,然后运行`source ~/.bashrc`(或重启)使更改生效。
记住,不同的cuDNN版本可能会有不同的安装步骤,确保遵循官方文档中的指导。
阅读全文