Ubuntu 20.04 安装cuda
时间: 2023-09-18 22:14:54 浏览: 172
在 Ubuntu 20.04 上安装 CUDA 可以按照以下步骤进行:
1. 下载 CUDA 安装包
在 NVIDIA 的官网上下载适合你的系统的 CUDA 安装包,下载地址为:https://developer.nvidia.com/cuda-downloads
选择好适合你的系统的版本之后,下载对应的 runfile 文件,比如 `cuda_11.2.2_460.32.03_linux.run`。
2. 禁用 Nouveau 驱动
在安装 CUDA 之前需要先禁用 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
```
3. 安装依赖
在安装 CUDA 之前需要安装一些依赖,可以通过以下命令进行安装:
```
sudo apt-get update
sudo apt-get install build-essential dkms freeglut3 freeglut3-dev libxi-dev libxmu-dev
```
4. 运行安装程序
在终端中进入到下载的 CUDA 安装包所在的目录,然后运行以下命令:
```
sudo sh cuda_11.2.2_460.32.03_linux.run
```
根据提示进行安装,需要注意的是,需要根据自己的需求选择不同的安装选项,比如是否安装驱动、是否安装 CUDA Toolkit、是否安装 CUDA 样例等等。
5. 配置环境变量
安装完成后,需要配置环境变量,可以在 `.bashrc` 文件中添加以下配置:
```
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
```
然后运行以下命令使配置生效:
```
source ~/.bashrc
```
6. 验证安装
最后,可以通过以下命令来验证 CUDA 是否安装成功:
```
nvcc -V
```
如果输出类似以下内容,则说明安装成功:
```
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Thu_Jul_15_19:09:01_PDT_2021
Cuda compilation tools, release 11.4, V11.4.100
Build cuda_11.4.r11.4/compiler.30188945_0
```
阅读全文