安装CUDA11.8 【Ubuntu 20.04]
1. 更改为清华源并更新软件列表和依赖项
https://mirrors.tuna.tsinghua.edu.cn/help/ubuntu/
# 默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-updates main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-backports main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-backports main restricted universe multiverse
# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-security main restricted universe multiverse
# # deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-security main restricted universe multiverse
deb http://security.ubuntu.com/ubuntu/ focal-security main restricted universe multiverse
# deb-src http://security.ubuntu.com/ubuntu/ focal-security main restricted universe multiverse
更新软件列表和检查必要的依赖项,如果没有,就安装
sudo apt-get update # 更新软件列表
sudo apt-get install g++ # 下载g++编译器
sudo apt-get install gcc # 下载gcc编译器
sudo apt-get install make # 下载GNU Make编译器
sudo apt-get install initramfs-tools # 下载安装initramfs-tools
2. 安装CUDA
步骤一: 下载CUDA安装包
进行CUDA和cuDNN的选择,也可以直接根据官方推荐进行下载安装。
从Nvidia官网下载CUDA https://developer.nvidia.com/cuda-downloads
下方链接,选择更多版本 https://developer.nvidia.com/cuda-toolkit-archive
- 接下来根据提示进行下载、安装
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run sudo sh cuda_11.8.0_520.61.05_linux.run
步骤二: 安装CUDA安装包过程中的选择
-
Do you accept the above EULA? (accept / decline / quit):
是否接受最终用户许可协议,输入accept -
回车键进行勾选,X就是选中,没有X就是没有选中,把安装驱动进行取消。之后向下键,回车确认
-
最后点击 install如下安装成功
步骤三: 配置CUDA环境
sudo vim ~/.bashrc
- 在bashrc文件最下方,添加下入代码
- (ps:这边需要注意cuda的版本,版本不同,路径的命名需修改)
export PATH=$PATH:/usr/local/cuda-11.8/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
- 更新环境
source ~/.bashrc
- 测试CUDA是否安装成功
nvcc -V
输出下述结果,表示安装成功:
wit622@622:~/cuda$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0
博客参考链接:【Ubuntu 20.04LTS系统】安装CUDA11.8、cuDNN,可进行CUDA版本切换_ubuntu_aglo-GitCode 开源社区 (csdn.net)