Linux下,Nvidia驱动以及Cuda的安装

最近刚好在看动手深度学习的书,需要在GPU环境下运行,故此记录一下。

一、确认安装环境

1.禁用nouveau(否则与NVIDIA驱动安装会有冲突,nouveau是linux自带系统驱动)

 右键打开终端
 sudo  vi /etc/modprobe.d/blacklist.conf(没有安装vim编辑器需要先执行 sudo apt-get install  vim)
 在打开的文件末尾添加以下两行代码
 blacklist nouveau
 options nouveau modeset=0
 保存退出
 update-initramfs –u   (重新生成 kernel initramfs)
 重启后(如果电脑有两个显卡驱动,图标不会变化;否则,图标会变大)打开终端输入一下命令:
 lsmod | grep nouveau (没有输出表示已经禁用)  

2.Nvidia驱动安装

A.执行如下命令将Ubuntu官方维护的NVIDIA显卡驱动软件源添加到系统中:

sudo add-apt-repository ppa:graphics-drivers/ppa 
sudo apt-get update

B.执行如下命令,查看适合自己电脑的显卡驱动,找到recommended对应项

sudo ubuntu-drivers devices

C找到电脑的设置>>软件和更新>>附加驱动>>选择对应的Nvidia驱动>>点击右下方按钮更改,完成后重启执行如下代码

nvidia-smi  测试驱动是否安装成功,成功则显示如下图:

Linux下,Nvidia驱动以及Cuda的安装_第1张图片

二、Cuda安装

1.下载与驱动对应的cuda的.run文件

Linux下,Nvidia驱动以及Cuda的安装_第2张图片
注意:下载好的.run文件移动到home目录下

2.安装

alt + ctrl + F1
sudo service lightdm stop     (表示在文字模式的安装cuda)
sudo  sh    cuda_8.0.61_375.26_linux.run/cuda_10.1.168_418.67_linux.run (自己下载好的文件名)

A 安装过程选项选择
由于自己在台式机和笔记本上分别安装cuda8.0和cuda10.1。
(1)对于cuda8.0会有一个很长的阅读声明,一直按住enter键直100%;而对于cuda10.1版本直接就是个图形化界面。阅读完成后,输入accept即可。
(2)cuda8.0中,选项install nvidia accelerated Graphics Driver,输入n,因为我们已经安装了nvidia的驱动;后面的选型均输入y,等待安装即可;
对于cuda10.1在它对应图形化界面取消该项,默认是【x】,ENTER键取消,选择install安装即可(为图形界面第一个选型)
说明:无论安装什么版本,我们都不安装Graphics Driver这一项即可(填n),其余选项均为 y
(3)如何最后看到有三行中,最后两行有两个installed,表示安装成功

B 输入reboot,重启电脑
C 相关配置

sudo vim ~/.bashrc  打开改文件

在文件最后添加以下两行:
export PATH=/usr/local/cuda-(对应版本号,如:10.1)/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-(同上)/lib64:$LD_LIBRARY_PATH12

保存后退到终端,执行如下命令:
source .bashrc
nvcc -V  检查是否安装成功

3测试

执行如下代码:
cd /usr/local/cuda/samples 
cd 1_Utilities/deviceQuery

sudo make
/deviceQuery

当看到Result=pass,则表示表示测试成功,如下图所示

在这里插入图片描述
参考相关博客:
https://www.linuxdashen.com/switch-intel-nvidia-graphics-card-ubuntu
https://www.linuxidc.com/Linux/2018-09/154292.htm

你可能感兴趣的:(Linux软件安装)