首页 > 系统服务 > 详细

Ubuntu18.04上安装CUDA_10.1(nvidia-driver)和cuDNN_7.6.5

时间:2021-01-28 17:37:51      阅读:30      评论:0      收藏:0      [点我收藏+]

https://www.cnblogs.com/old-sister/p/14000893.html

本文是在Ubuntu18.04.5服务器上安装CUDA_10.1(nvidia-driver455)和cuDNN_7.6.5,

  • Ubuntu 18.04.5
  • CUDA_10.1 (nvidia-driver455)
  • cuDNN_7.6.5

一、 前期准备

1、查看系统版本和显卡型号

1
2
3
cat /etc/issue # 查看系统版本
 
lspci | grep -i nvidia # 查看电脑的显卡型号

技术分享图片


2、查看系统是否安装显卡驱动 nvidia-driver

1
nvidia-smi # 查看显卡驱动情况(此处未安装,应无信息显示)

3、显卡支持的驱动型号

1
ubuntu-drivers devices # 查看显卡支持的驱动型号

技术分享图片

此处可以看到系统推荐的驱动版本是 455:

4、安装推荐版本455的显卡驱动

1
sudo ubuntu-drivers autoinstall # 安装默认显卡驱动(此处为455)  

5、重启服务器(无其它用户使用服务器)

1
2
3
(sudo) reboot # 立即重启
(sudo) shutdown -r now # 立即重启  

6、查看显卡驱动是否安装成功:

1
nvidia-smi

 

安装成功会显示
技术分享图片

7、卸载已有驱动程序(需要重启才生效)

1
2
3
$ sudo apt-get remove nvidia-*
$ sudo apt-get remove --purge nvidia*

二、 RUN方式安装CUDA _10.1 (我有安装过11.0,11.1,后因项目需求又改为10.1,图片有点混)

0、安装之前需要确认已安装gcc和g++

1
2
$ gcc --version
$ g++ --version

1、[官网cuda_10.1 下载地址]( https://developer.nvidia.com/cuda-10.1-download-archive-base?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal ):

2、点击下载:cuda_10.1.105_418.39_linux.run
技术分享图片

3、上传cuda_10.1.105_418.39_linux.run文件至ubuntu系统上的 home/your_username目录下

4、执行命令安装

1
sudo sh cuda_10.1.105_418.39_linux.run

5、同意协议:输入accept
技术分享图片

6、此处选择是否安装nvidia-driver (回车切换是否选择,我已单独安装nvidia-driver,所以此处选择不安装驱动)

最后选择 install
技术分享图片

7、安装成功

技术分享图片

8、环境配置(两种方式)

8.1、~.bashrc # 修改home目录下的.bashrc文件,只针对当前用户

安装完后,在.bashrc文件末尾添加环境变量

sudo vim ~/.bashrc
  • 1
export PATH=$PATH:/usr/local/cuda/bin  
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64  
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64  
  • 1
  • 2
  • 3
保存退出后,输入以下命名
source ~/.bashrc

  • 1
  • 2
  • 3

测试是否成功

sudo rm -rf /usr/local/cuda  
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda  
nvcc --version  

 cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

这样的结果就ok 下边有个pass
技术分享图片
在去官网找linux 的cudnn https://developer.nvidia.com/rdp/cudnn-download

Ubuntu18.04上安装CUDA_10.1(nvidia-driver)和cuDNN_7.6.5

原文:https://www.cnblogs.com/shuimuqingyang/p/14338042.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!