ubuntu安装cuda、cudnn
时间: 2023-09-12 15:02:35 浏览: 133
安装CUDA和cuDNN是在Ubuntu上进行深度学习开发的常见任务。下面是安装CUDA和cuDNN的一般步骤:
1. 安装NVIDIA显卡驱动程序:
- 打开终端,运行以下命令以添加NVIDIA驱动PPA存储库:
```
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
```
- 运行以下命令来安装最新的NVIDIA显卡驱动程序:
```
sudo apt install nvidia-driver-<version>
```
(<version> 是您安装的驱动程序版本,例如:nvidia-driver-460)
2. 安装CUDA Toolkit:
- 访问NVIDIA官方网站(https://developer.nvidia.com/cuda-toolkit-archive)并下载适用于您的操作系统的CUDA Toolkit安装程序。
- 运行下载的安装程序,并按照提示进行安装。确保在安装过程中选择合适的选项。
3. 配置环境变量:
- 打开终端,编辑`~/.bashrc`文件:
```
nano ~/.bashrc
```
- 在文件末尾添加以下内容(根据您的CUDA版本进行调整):
```
export PATH=/usr/local/cuda-<version>/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-<version>/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
(<version> 是您安装的CUDA版本,例如:cuda-11.0)
4. 安装cuDNN:
- 访问NVIDIA官方网站(https://developer.nvidia.com/rdp/cudnn-download)并下载适用于您的CUDA版本的cuDNN库。
- 解压下载的文件,并将文件复制到CUDA安装目录:
```
sudo cp cuda/include/cudnn*.h /usr/local/cuda-<version>/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-<version>/lib64
sudo chmod a+r /usr/local/cuda-<version>/include/cudnn*.h /usr/local/cuda-<version>/lib64/libcudnn*
```
5. 更新环境变量:
- 运行以下命令以使环境变量生效:
```
source ~/.bashrc
```
完成上述步骤后,您应该已经成功安装了CUDA和cuDNN。您可以运行`nvcc --version`命令来验证CUDA是否安装正确。
阅读全文
相关推荐

















