安装显卡驱动
打开‘软件和更新。
点击附加驱动安装显卡驱动
如果已经安装显卡驱动,请忽略上面的步骤。
安装gcc
新安装的Ubuntu22.04 没有安装gcc,需要安装gcc。在终端输入gcc -version
查看有没有gcc。
执行命令
sudo apt install gcc
安装CUDA
输入nvidia-smi
查看支持CUDA的版本,通过下图得知,支持的最高版本是11.8.
我在这里选中11.7的版本,主要因为pytorch目前支持的最高版本是11.7.
CUDA下载连接:https://developer.nvidia.com/cuda-11-7-1-download-archive
选择对应的版本,然后在下面就可以安装命令:
wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run
sudo sh cuda_11.7.1_515.65.01_linux.run
下载完成后,执行sudo sh cuda_11.7.1_515.65.01_linux.run
选择Continue
。
输入accept
。
点击空格,取消安装驱动,然后选择
Install
。
看到上面的消息,就说明安装完成了。
配置环境变量
输入gedit ~/.bashrc
命令打开文件,在文件结尾输入以下语句,保存。
export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
然后,点击保存后关闭文档,然后执行命令source ~/.bashrc
。
至此cuda安装完成,输入nvcc -V
命令查看cuda信息。