当前位置:网站首页 > 经验分享 > 正文

ubuntu20安装nvidia显卡驱动

这是我参与8月更文挑战的第7天,活动详情查看:8月更文挑战

软硬件环境

  • ubuntu 18.04 64bit
  • NVidia GTX 1070
  • CUDA 10.1
  • cudnn 7.6.3

安装nvidia驱动

进入没有安装驱动的ubuntu系统,屏幕显示是不正常的,分辨率偏小。从官网下载RUN文件,比如我这里的NVIDIA-Linux-x86_64-435.21.run,这里的版本选择需要与后面的cuda进行匹配

# 添加执行权限 chmod a+x NVIDIA-Linux-x86_64-435.21.run sudo ./NVIDIA-Linux-x86_64-435.21.run 

安装过程基本上就是选择yes或者no,装完后重启系统,就会发现屏幕显示正常了,分辨率也变高了。

由于cuda的安装文件中已经包含了对应的驱动,所以,这一步,其实可以不用先做,当然了,这样安装也没有问题,还是看个人习惯

安装CUDA

接下来开始安装cuda,官方下载地址: developer.nvidia.com/cuda-downlo…

# 添加可执行的权限 chmod +x cuda_10.1.243_418.87.00_linux.run sudo ./cuda_10.1.243_418.87.00_linux.run 

安装过程中会有一些选项出现,根据自己的实际情况选择,比如要不要使用cuda自带的驱动,安装路径、是否安装samples等(每个版本不太一样)

ubuntu cuda

默认情况下,cuda被安装在了/usr/local下,同时还创建了一个软链接cuda

ubuntu cuda

然后配置下环境变量,编辑~/.bashrc文件

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

最后执行下source

source ~/.bashrc 

这样几个重要的环境变量就立马生效了。或者,重新打开terminal,也同样生效。

安装cudnn

到官网 developer.nvidia.com/cudnn 去下载,这里需要注册,有点麻烦。需要注意的是,cudnn的版本是要跟cuda匹配的

tar xvf cudnn-10.1-linux-x64-v7.6.3.36.tgz sudo cp cuda/include/cudnn*.h /usr/local/cuda-10.1/include sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64 # 添加读的权限 sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn* 

测试

安装完毕,我们来测试下环境是否都ok?

进入到sample中,跑下deviceQuery

# 安装编译工具 sudo apt install build-essential cd NVIDIA_CUDA-10.1_Samples/1_Utilities/deviceQuery make 

ubuntu cuda

最后执行nvidia-smi看看

ubuntu cuda

刚好我这里跑了示例,所以你能看到gpu基本上是满负荷工作

多cuda版本共存及切换

cuda的安装就可以看到,多个版本的cuda共存是可行的,而且切换也非常简单。

安装过程跟上面提到的是一样的,有几个版本就安装几个,比如我说这里安装了2个,一个10.0,一个10.1,且当前激活的是10.1版本

ubuntu cuda

现在我们来操作下切换

# 删除软链接 sudo unlink /usr/local/cuda # 重新创建软链接,以使用10.1为例 sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda 

参考资料

  • www.nvidia.com/Download/in…
  • developer.nvidia.com/cuda-toolki…

版权声明


相关文章:

  • ubuntu 16.04安装nvidia显卡驱动2025-07-31 11:01:05
  • 观测仪图片2025-07-31 11:01:05
  • 图片轮播特效代码2025-07-31 11:01:05
  • java技术栈思维导图2025-07-31 11:01:05
  • java编写软件2025-07-31 11:01:05
  • threejs 移动2025-07-31 11:01:05
  • java实现AES的五种加密模式2025-07-31 11:01:05
  • junit.vintage2025-07-31 11:01:05
  • 有源晶振和无源晶振哪个好2025-07-31 11:01:05
  • 谈谈跨域问题2025-07-31 11:01:05