linux nvidia显卡驱动安装 cuda cudnn
时间: 2025-03-15 17:20:42 AIGC 浏览: 77
### Linux 下 NVIDIA 显卡驱动、CUDA 和 cuDNN 的安装教程
#### 1. 安装 NVIDIA 显卡驱动
在 Ubuntu 上安装 NVIDIA 显卡驱动可以通过图形界面完成,也可以通过命令行操作。推荐方法如下:
- 打开 *软件和更新* 中的 *附加驱动* 页面,在可用选项中选择带有 `NVIDIA` 字样的驱动程序并应用更改[^2]。
如果需要手动安装或者禁用默认的 Nouveau 驱动,则可以执行以下步骤:
```bash
sudo apt update
sudo apt install linux-headers-$(uname -r)
sudo modprobe -r nouveau && sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nouveau.conf"
sudo reboot
```
重启后下载对应版本的 NVIDIA 驱动包,并运行安装脚本[^3]:
```bash
chmod +x NVIDIA-Linux-x86_64-version.run
sudo ./NVIDIA-Linux-x86_64-version.run
```
#### 2. 升级或安装 CUDA 工具包
为了确保兼容性和性能优化,建议先确认已安装的 NVIDIA 驱动版本是否支持目标 CUDA 版本。
对于特定版本如 CUDA 10.1 可以按照官方文档说明进行配置[^1]。通常情况下可通过 APT 或者本地 RUN 文件两种方式实现安装过程。APT 方法较为简便:
```bash
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntuXX/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-10-1
```
完成后记得设置环境变量以便后续调用工具链正常工作:
```bash
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH
source ~/.bashrc
```
#### 3. 配置 cuDNN 库文件
cuDNN 是针对深度学习框架加速而设计的一套高性能库集合。其依赖于基础 CUDA 平台之上构建而成。获取合法授权后的 cuDNN 压缩档需解压到指定目录下覆盖原有内容:
假设当前路径存在 tar.gz 形式的压缩包形式:
```bash
tar zxvf cudnn-X-linux-x64-vY.Y.ZZ.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
```
最后验证整个流程无误可尝试编译示例项目来检测功能完整性。
```python
import tensorflow as tf
print(tf.test.is_built_with_cuda()) # 输出 True 表明成功启用GPU计算能力
```
阅读全文
相关推荐



















