ubuntu 20 安装 CUDA

1. 查看需要安装的cuda版本

nvidia-smi

cuda的版本信息如下图所示 

ubuntu 20 安装 CUDA_第1张图片

2. 去官网下载对应版本的CUDA 

官网:CUDA Toolkit Archive | NVIDIA Developer

ubuntu 20 安装 CUDA_第2张图片 

弹出以下界面,依次点击以下按钮

ubuntu 20 安装 CUDA_第3张图片

 得到以下内容:

ubuntu 20 安装 CUDA_第4张图片复制下载链接,下载cuda11到本地

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run

3. 安装cuda

1. ctrl+alt+F1, 退出图形界面,进入命令行界面

2. 关闭lightdm

sudo service lightdm stop

3. 通过命令行安装cuda

sudo sh cuda_11.4.0_470.42.01_linux.run

        注意事项:因为之前已经安装过nvidia驱动了,这里不再安装NVIDIA显卡驱动

ubuntu 20 安装 CUDA_第5张图片

最后选择Install,按enter开始安装。

见到以下界面,安装完毕。

ubuntu 20 安装 CUDA_第6张图片

 4. 重新打开图形界面。最好重启计算机

sudo service lightdm restart
sudo reboot

5. 配置cuda的环境路径

sudo vim ~/.bashrc

在末尾加入以下内容:

export PATH=/usr/local/cuda-11.4/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH

关闭.bashrc文件,并更新至应用

source ~/.bashrc

6. 验证更新成功

新建一个terminal 

nvcc -V

你可能感兴趣的:(ubuntu,深度学习,linux)