Ubuntu 20.04 & CUDA

本文提供了一步一步的指导,教你如何在Ubuntu20.04上取消默认Nouveau驱动,安装NVIDIA驱动,并配置CUDA环境,包括编译环境的搭建和CUDA的安装。

Ubuntu 20.04 & CUDA

注意:部分sudo操作需要su - 为root。

取消默认驱动Nouveau

$ sudo echo "blacklist nouveau" >>/etc/modprobe.d/blacklist.conf
$ sudo echo "options nouveau modeset=0" >>/etc/modprobe.d/blacklist.conf
$ sudo update-initramfs -u
$ sudo reboot

安装编译环境

$ sudo apt install build-essential

安装驱动

$ sudo bash NVIDIA-Linux-x86_64-440.82.run

安装CUDA

### 如何在 Ubuntu 20.04 上安装 CUDA Toolkit #### 准备工作 确保已正确安装适用于 NVIDIA 显卡的驱动程序[^4]。这是成功安装 CUDA 的前提。 #### 添加 NVIDIA 软件源并更新包列表 为了简化后续操作,建议先配置官方软件仓库: ```bash sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update ``` #### 下载与安装 CUDA Toolkit 有两种主要方式来获取和部署 CUDA 工具链:通过 `.run` 文件 或者 使用 APT 包管理器。推荐采用更稳定的APT方式进行安装: ##### 方法一:利用 APT 进行安装 (推荐) 1. 注册 NVIDIA GPG 密钥库: ```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 ``` 2. 更新本地包索引并将 CUDA 存储库添加到系统中 ```bash sudo apt-get update ``` 3. 开始安装特定版本的 CUDA,比如这里以安装 `cuda-11-0` 版本为例 ```bash sudo apt install cuda-11-0 ``` 4. 设置环境变量以便命令行可以直接调用 nvcc 编译器等工具 将下面两行加入用户的 shell 配置文件(`~/.bashrc`, `~/.zshrc`)末尾处: ```bash export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64\ ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 5. 让更改生效 ```bash source ~/.bashrc # 如果使用的是 bash # 或者对于 zsh 用户来说应该是这个 source ~/.zshrc ``` 此时应该可以验证安装是否成功的状态了. #### 验证安装结果 可以通过运行如下命令测试编译器是否正常工作以及查看当前使用的 CUDA 版本号: ```bash nvcc --version ``` 如果一切顺利的话,则会显示所安装的具体版本信息。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值