1、使用命令行安装
sudo apt-get install nvidia-driver-515
注意:
(1)、有些GPU显卡只支持455驱动,如果515安装失败,可以尝试安装455显卡驱动。
(2)、安装结束后,重启环境:sudo reboot
2、验证安装是否成功
命令行输入:nvidia-smi,提示如下:
3、安装CUDA
下载地址:CUDA Toolkit 11.0 Update 3 Downloads
选择Linux -> x86_64 -> Ubuntu -> 20.04 -> runfile[local]
安装命令:
wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
等待下载完成:
CUDA下载完成后,安装即可:
sudo sh cuda_11.0.3_450.51.06_linux.run
键盘上下按钮,选择continue:
鼠标悬停处输入:accept,弹出如下也页面:
注意:由于前面我们已经安装了NVIDIA驱动,在此需要将Driver去掉,不进行安装,使用键盘上下按钮选择Driver,点击enter按钮,去掉选中。
如图所示,使用上下按钮,选中install,点击enter按钮,开始进行安装。
安装成功后:
4、添加环境变量:
打开配置文件:sudo vim ~/.bashrc,在配置文件最后添加下列路径:
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=${LD_LIBARY_PATH}:${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}
保存并关闭,然后使用命令:source ~/.bashrc将配置生效。
查看CUDA是否安装成功,输入命令:nvcc -V:
总计
总结:到此CUDA安装成功。