ubuntu下安装cuda的两种方法

1. 安装方法1

以安装cuda9.0为例,首先在CUDA Toolkit Archive网页找到CUDA Toolkit 9.0版本,然后选择如下选项:

ubuntu下安装cuda的两种方法_第1张图片

对。其实底部已经有安装说明了,这里再详细说明一下。
首先肯定是将.deb安装包下载到本地。
然后使用ctrl+alt+F1进入终端,使用如下命令关掉图形用户界面:

sudo service lightdm stop

然后卸载所有之前安装的nvidia驱动以及cuda相关文件:

sudo apt remove --purge nvidia*
sudo apt remove --purge cuda*

再检查 /usr/local/下面是否还存在cuda相关的文件夹,因为cuda默认是被安装到 /usr/local/ 路径下的。如果有的话,使用sudo rm 命令将其残留的文件夹全部删除。

现在cd到.deb包所在的文件夹,然后执行上图中的第一条dpkg命令:

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb

执行完之后,在路径 /etc/apt/source.list.d/ 下会生成一个cuda-9-0-local.list文件:

user@intsig-PC01012:~$ cd /etc/apt/sources.list.d/
user@intsig-PC01012:/etc/apt/sources.list.d$ ll
total 28
drwxr-xr-x 2 root root 4096 5月  22 14:52 ./
drwxr-xr-x 7 root root 4096 5月  22 14:20 ../
-rw-r--r-- 1 root root   38 9月   2  2017 cuda-9-0-local.list
-rw-r--r-- 1 root root  189 6月  25  2018 google-chrome.list
-rw-r--r-- 1 root root  189 6月  25  2018 google-chrome.list.save
-rw-r--r-- 1 root root   65 6月  25  2018 sogoupinyin.list
-rw-r--r-- 1 root root   65 6月  25  2018 sogoupinyin.list.save
user@intsig-PC01012:/etc/apt/sources.list.d$

有经验的同学知道,在目录 /etc/apt/ 下面存在两个文件:

sources.list
sources.list.d/

首先明确一点:所有的.list文件中都存放着apt安装源。
其中,sources.list里面配置了官方的apt安装源路径,即我们平时使用sudo apt install...命令安装软件时,系统都是在sources.list文件中提供的源中寻找可用的安装包,然后下载到本地进行安装的。
另一个sources.list.d/文件夹下据说是用来存放第三方安装源。比如刚才我们执行dpkg命令,实际上就是新建了一个本地安装源,安装源就存放在生成的

/etc/apt/sources.list.d/cuda-9-0-local.list

文件中,现在打开该文件看一下本地安装包被放在哪儿:

user@intsig-PC01012:/etc/apt/sources.list.d$ cat cuda-9-0-local.list 
deb file:///var/cuda-repo-9-0-local /
user@intsig-PC01012:/etc/apt/sources.list.d$ 

可以看到,本地cuda仓库被放到了

/var/cuda-repo-9-0-local/

现在,执行

sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt update

本地cuda仓库就能被apt命令检测到了,最后执行

sudo apt install cuda

没问题的话,cuda9.0就安装结束了。为了验证cuda是否安装成功,执行

nvidia-smi

命令,没问题的话应该会弹出GPU的相关信息。
最后再打开图形桌面:

sudo service lightdm start

cuda安装完毕。

2. 安装方法2

第二种安装方式超级简单。
前面说过,apt第三方安装源是被配置到 /etc/apt/sources.list.d/ 文件夹下的.list文件中的。因此,我们可以直接新建一个apt安装源,比如新建一个cuda.list文件:

sudo vim /etc/apt/sources.list.d/cuda.list

在里面加入:

deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /

执行 :x 保存文件,然后执行sudo apt update命令更新apt源。
现在你就可以使用apt命令安装cuda了,而且可以安装你想要的任何版本,输入

sudo apt install cuda-

按两下Tab键,就会显示可供安装的各个cuda版本,8.0、9.0、10.0等等,你想装哪个版本都行。如果要安装cuda9.0,直接执行

sudo apt install cuda-9-0

简单吧! 当然,同样需要先进入终端、关掉图形用户桌面、清理之前安装的nvidia驱动和cuda文件等等,操作过程和方法1一样。

3. 总结

正常情况下,使用方法2安装cuda是最简单的,但是方法2组要从网上服务器在线下载安装包。如果网速不好就GG了,这时就用第一种方法吧。

你可能感兴趣的:(ubuntu下安装cuda的两种方法)