ubuntu18.04安装NVIDIA驱动+CUDA10

实验室做GPU实验,需要用到NVIDIA显卡+ CUDA10,安装过程中走了很多坑,在此记录。

  1. 查看本地显卡
XXXXX:~$ lspci | grep VGA
01:00.0 VGA compatible controller: NVIDIA Corporation GP104M [GeForce GTX 1070 Mobile] (rev a1)

可以看到我的显卡是:GeForce GTX 1070的, 如果没有说明显卡没插好或者其他问题,先解决。

  1. 查看显卡驱动
xxxx:~$  lshw -c video  # 看configurure字段有木有driver字样,若有内容,则表示对应的工作站显卡驱动装好了。

## 但是!!! Ubuntu18.04 或者其他很有可能是 nouveau ,此次实验室做深度学习,要用到NVIDIA的,因此需要改显卡驱动,先删除旧的,再禁用nouveau;

ubuntu18.04安装NVIDIA驱动+CUDA10_第1张图片
   2.1 删除原有驱动

sudo apt-get remove --purge nvidia*  # 其实这也不删问题也不大

   2.2 禁用nouveau驱动
 修改blacklist.conf文件,添加 blacklist nouveau,具体操作如下:

sudo vim /etc/modprobe.d/blacklist.conf
blacklist nouveau  #添加数据
  1. 选择合适显卡,安装显卡
    3.1 下载显卡驱动
    建议前往官网下载,选择对应型号的驱动下载,我下载的是NVIDIA-Linux-x86_64-418.43.run
    ubuntu18.04安装NVIDIA驱动+CUDA10_第2张图片
    3.2安装显卡驱动
    下面安装显卡操作建议在命令行进行,安装显卡可能会出现桌面显示不了的问题,倒不如之间在命令行操作,进入命令行操作,进入系统时按:Ctrl+Alt+[ F1 -F6] , F1-F6任选一个,F7是进入图形界面,这里不用;

安装前先更新下配置,重启按Ctrl+Alt+ F1,进入命令行

sudo update-initramfs -u  # 更新下配置
reboot  # 进入系统前 按Ctrl+Alt+ F1 进入命令行

在命令行找到已经下载好的文件,运行如下命令安装:

sudo bash NVIDIA-Linux-x86_64-418.43.run

显卡查看,出现如下请说明显卡已正常安装,其中 Volatile项是使用率。

nvidia-smi

ubuntu18.04安装NVIDIA驱动+CUDA10_第3张图片
4. 安装CUDA10
建议前往官网下载选择对应的版本下载并安装。我选择runfile方式安装,文件比较大建议下载到本地再安装,在线安装容易出错,或者改成清华源进行安装,速度快些。
ubuntu18.04安装NVIDIA驱动+CUDA10_第4张图片
我下载的是: cuda_10.1.105_418.39_linux.run, 如下安装

sudo sh   cuda_10.1.105_418.39_linux.run

安装过程中,驱动前面已经安装,就不需要了,其他的默认安装,请注意,X 表示安装,空表示不安装。
ubuntu18.04安装NVIDIA驱动+CUDA10_第5张图片
添加环境变量

vim  ~/.bashrc  # 进入 .bashrc文件中,在最后添加如下三行代码
	export CUDA_HOME=/usr/local/cuda 
	export PATH=$PATH:$CUDA_HOME/bin 
	export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc  # 保存并退出
  1. 查看CUDA
nvcc -V   # 如下图可以看到已经安装成功CUDA10.1版本了。

ubuntu18.04安装NVIDIA驱动+CUDA10_第6张图片

你可能感兴趣的:(pytorch)