ubuntu cuda cudnn安装
时间: 2025-01-09 15:51:34 浏览: 4
### 如何在Ubuntu上安装CUDA和cuDNN
#### 准备工作
为了确保顺利安装,在开始之前应当移除任何可能存在的旧版CUDA工具包以及PPA源。这一步骤可以通过执行如下命令完成:
```bash
sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove nvidia-cuda-toolkit
```
上述操作可以防止新旧版本冲突,从而保障系统的稳定性[^2]。
#### 获取适合的CUDA版本
通过`nvidia-smi`命令查看当前使用的NVIDIA显卡能够支持的最大CUDA版本号是非常重要的。这是因为如果尝试安装超过此限度的CUDA版本,则可能导致兼容性问题。因此建议访问[CUDA Toolkit Archive](https://developer.nvidia.com/cuda-toolkit-archive),从中挑选合适的CUDA版本进行下载并遵循官方文档中的指导来完成安装过程[^1]。
#### 安装cuDNN库
一旦成功安装了所需的CUDA环境之后,下一步就是配置cuDNN。通常情况下,可以从[NVIDIA cuDNN官网](https://developer.nvidia.com/rdp/cudnn-download)获取适用于已安装CUDA版本的cuDNN文件,并将其解压到相应的目录下(通常是`/usr/local/cuda/include` 和 `/usr/local/cuda/lib64`)。完成后记得设置好环境变量以便程序能正确调用这些库函数。
#### 测试安装成果
最后但同样重要的是验证整个流程是否无误。一种简单的方法是在终端里输入`nvcc --version`检查CUDA编译器的存在及其版本信息;另外也可以运行一些简单的测试样例比如`deviceQuery` 或者 `bandwidthTest` 来进一步确认GPU加速功能正常运作。
阅读全文