ubuntu cuda和cudnn安装
时间: 2025-01-17 18:51:12 浏览: 30
### 安装准备
为了确保顺利安装CUDA和cuDNN,在Ubuntu上需先确认系统的兼容性和现有软件包的状态。这包括检查操作系统版本、显卡信息及其驱动状态。
对于操作系统的检测,可以利用命令`uname -m && cat /etc/*release*`来获取架构详情与发行版信息[^4]。而针对显卡的信息收集,则推荐使用`lspci | grep -i nvidia`以识别NVIDIA GPU的存在,并通过`nvidia-smi`进一步了解当前激活的驱动程序版本及性能概况[^1]。
### 驱动安装
在着手于CUDA之前,保证有适当版本的NVIDIA驱动被正确部署至关重要。通常建议采用官方源中的稳定版或是直接从制造商处下载适用于特定硬件的新近发布版本。执行如下指令序列可实现这一点:
```bash
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
ubuntu-driver autoinstall
```
上述过程会自动挑选并应用最适合本地设备状况的选择之一[^3]。
### CUDA Toolkit 的安装
当基础环境搭建完毕之后,便能转向核心组件——即CUDA工具集本身的引入工作上来。依据个人需求选取相匹配的主要分支(比如11.x系列),接着遵循下面给出的操作流程来进行实际部署:
#### 方法一:借助APT仓库简化流程
这种方法适合追求便捷性的用户群体。只需依次运行下列语句即可达成目标:
```bash
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda
```
#### 方法二:手动处理.run 文件
如果偏好更精细控制或遇到某些特殊情况时,可以从官方网站获取独立的`.run`脚本形式分发包。不过在此之前务必停用任何现存的图形界面服务以免引起冲突;另外记得按照提示逐步完成交互式向导直至结束。
无论采取哪种途径,最终都应重启计算机使更改生效,并可通过`nvcc --version`检验是否达到预期效果[^2]。
### cuDNN 库集成
有了稳固的基础平台支持后,下一步就是增添专门面向深度神经网络优化过的数学运算库——cuDNN了。鉴于其紧密依赖于先前建立起来的CUDA框架之上,因此务必要严格对照两者之间的协同关系表单做出恰当选择。
假设已经取得了对应压缩包,那么继续参照下述指导方针推进剩余环节:
```bash
tar xvzf cudnn-linux-x86_64-*.*.*_*.*
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64' >> ~/.bashrc
source ~/.bashrc
```
至此,整个设置链条宣告完工。此时不妨尝试编译一段简单的测试案例加以验证,例如MNIST手写数字分类器训练作业等,从而直观感受GPU带来的效率提升优势。
阅读全文
相关推荐


















