Ubuntu16.04:NVIDIA cuda10.0安装

背景

之前曾说过使用sudo ubuntu-drivers devices来安装显卡驱动,这样可以保证基本上不出大问题。然而世界上的事情都是舍不得孩子套不住狼;使用这种默认方法安装的驱动版本太低,仅仅能安装cuda8.0,实在难以想象。所以我决定重新安装一次ubuntu系统,然后从头安装一次驱动,并按照cuda10。
参考教材:
https://www.cnblogs.com/fanminhao/p/8902296.html
http://www.cnblogs.com/fanminhao/p/8902378.html
https://www.cnblogs.com/gambler/p/9574596.html


流程

1、安装ubuntu后更新

非常重要!!!

sudo apt-get update
sudo apt-get upgrade

2、更新Ubuntu Kernel

在更新kernel前,首先要安装libssl1.1.0,这个链接如下:
百度网盘地址
https://pan.baidu.com/s/155rYJB2NnAvGOW65zVpCvQ 密码: 5mdr

由于我的电脑的amd64,所以从https://kernel.ubuntu.com/~kernel-ppa/mainline/上选择一个比较新的Kernel,本次选择的是kernel v4.20,下载以下文件并放到一个单独文件夹(不能保证其他核也行):

  linux-headers-4.20.0-042000_4.20.0-042000.201812232030_all.deb
  linux-headers-4.20.0-042000-generic_4.20.0-042000.201812232030_amd64.deb
  linux-image-unsigned-4.20.0-042000-generic_4.20.0-042000.201812232030_amd64.deb
  linux-modules-4.20.0-042000-generic_4.20.0-042000.201812232030_amd64.deb

最后通过sudo dpkg -i *.deb安装新的内核,最后重启后通过uname -sr查看内核是否安装成功。

3、NVIDIA最新驱动

首先屏蔽nouveau驱动

sudo chmod 666 /etc/modprobe.d/blacklist.conf
sudo gedit /etc/modprobe.d/blacklist.conf

添加

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb

更新blacklist.conf文件

 sudo update-initramfs -u

关闭图像界面,准备安装显卡。
此处需要额外注意,ubuntu16.04由于gcc/g++版本较低,所以配合CUDA10.0,建议使用418.30。目前比较新版本的驱动,几乎都不能使用。
另外需要注意的是,驱动版本必须满足CUDA的要求,如下:
在这里插入图片描述

sudo service lightdm stop
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.run
sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau-check -no-opengl-files
sudo service lightdm start

使用nvidia-smi检查驱动是否安装成功。

4、安装CUDA10和cudnn

4.1 安装CUDA10.0

sudo chmod a+x cuda_10.0.130_410.48_linux.run
sudo ./cuda_10.0.130_410.48_linux.run

安装时一定要在安装驱动时,选择no,打开图像界面
此时一般nvcc还没有加入全局环境,打开sudo gedit ~/.bashrc添加

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

然后再source ~/.bashrc后,即可用nvcc --version看版本信息。

4.2 安装cudnn

解压压缩包后,执行以下命令


sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

5、验证

最后选取一个cuda程序验证一下,本文使用了https://github.com/Yannnnnnnnnnnn/cuda_cmakelists.git测试CUDA是否正常。

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 撸撸猫 设计师:C马雯娟 返回首页