Ubuntu20.04安装GPU显卡驱动

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安装成功。 

猜你喜欢

转载自blog.csdn.net/weixin_38863607/article/details/132127814