温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

Ubuntu上如何安装CUDA

发布时间:2022-10-27 09:26:06 来源:亿速云 阅读:486 作者:iii 栏目:服务器

本篇内容介绍了“Ubuntu上如何安装CUDA”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!

安装过程:

一、安装、熟悉ubuntu16.04系统

安装软件之前最好对ubuntu的命令行有一些基本的了解,像sudo、cd、ls、nona、cat、chmod等等,这样能够在安装过程中省去很多不必要的麻烦。

二、检查自己的电脑环境是否具备安装cuda的条件

1) 验证自己的电脑是否有一个可以支持cuda的gpu

你可以电脑的配置信息中找到显卡的具体型号,如果你是双系统,在windows下的设备管理器中也可以查到显卡的详细信息;
你也可以在ubuntu的终端中输入命令: $ lspci | grep -i nvidia ,会显示出你的nvidia gpu版本信息,不过不是很详细。

我的显示为(geforcegt630m):

01:00.0 3d controller: nvidia corporation gf117m [geforce 610m/710m/810m/820m / gt 620m/625m/630m/720m] (rev a1)

然后去cuda的查看自己的gpu版本是否在cuda的支持列表中。

2) 验证自己的linux版本是否支持 cuda(ubuntu 16.04没问题)

输入命令:

$ uname -m && cat /etc/*release

结果显示:

x86_64
distrib_id=ubuntu
distrib_release=16.04

......

3) 验证系统是否安装了gcc

在终端中输入: $  gcc –version     

结果显示:

gcc (ubuntu 5.4.0-6ubuntu1~16.04.5) 5.4.0 20160609
......

若未安装请使用下列命令进行安装:

sudo apt-get install build-essential

4) 验证系统是否安装了kernel header和 package development

a、查看正在运行的系统内核版本:

在终端中输入: $  uname –r

结果显示:

4.10.0-40-generic

b、在终端中输入:$  sudo apt-get install linux-headers-$(uname -r)

可以安装对应kernel版本的kernel header和package development

结果显示:

......

升级了 0 个软件包,新安装了 0 个软件包,要卸载 0 个软件包,有 4 个软件包未被升级。

表示系统里已经有了,不用重复安装。

若以上各项验证检查均满足要求,便可进行下面的正式安装过程。如果没有满足要求的话,可以参考cuda的官方文档,里面有详细的针对每个问题的解决方案。

三、选择安装方式

cuda提供两种安装方式:package manager安装和runfile安装, package manager 安装方式相对简单一些,但是我在阅读别人博客的过程中发现选择这种方式在安装过程中问题可能多一点,失败的概率较大。为了减少不必要的麻烦我选择runfile安装方式。
下载cuda安装包:cuda官网下载,根据系统信息选择对应的版本,runfile安装的话最后一项要选择 runfile文件。

下载完后,用md5 检验,如果序号不和,得重新下载(由于当时没有保存,这里我采用了别人的截图,注意这里的cuda版本号,人家采用的cuda_8.0)

输入命令:$  md5sum cuda_9.0.176_linux.run

Ubuntu上如何安装CUDA

四、runfile安装cuda

1) 禁用 nouveau驱动

终端中运行:$  lsmod | grep nouveau,如果有输出则代表nouveau正在加载。需要我们手动禁掉nouveau。

ubuntu的nouveau禁用方法:

a、在/etc/modprobe.d中创建文件blacklist-nouveau.conf

输入命令:$  sudo vi /etc/modprobe.d/blacklist-nouveau.conf (利用vi编辑器编辑和保存文件)

在文件中输入一下内容:

blacklist nouveau
options nouveau modeset=0

b、执行:

$ sudo update-initramfs –u

c、再执行:

$ lsmod | grep nouveau

若无内容输出,则禁用成功,若仍有内容输出,请检查操作,并重复上述操作。

注:vi是linux终端下或控制台下常用的编辑器,基本的操作方式为:vi /路径/文件名
例如:vi /etc/fstab表示显示/etc/fstab文件的内容。使用键盘上的page up和page down键可以上下翻页;按下insert键,可以见到窗口左下角有“insert”字样,表示当前为插入编辑状态,这时从键盘输入的内容将插入到光标位置;再按下insert键,左下角将有“replace”字样,表示当前为替换编辑状态,这时从键盘输入的内容将替换光标位置的内容。编辑完内容后,按下esc键,并输入“:wq”,然后回车就可以保存退出。
如果不想保存而直接退出,则按下esc键后,输入“:q!”,然后回车即可。“wq”表示write和quit,即保存退出;“q!”表示忽略修改强行退出。

以下操作建议手机拍照,建议把下载的cuda_9.0.176_384.81_linux.run文件重命名为cuda.run并移动到home文件夹下(为了安装方便)

2) 重启电脑,进入登录界面的时候,不要登录进入桌面(否则可能会失败,若不小心进入,请重启电脑),直接按ctrl+alt+f1进入文本模式(命令行界面),登录账户。

3) 输入 $ sudo service lightdm stop 关闭图形化界面

4) 切换到cuda安装文件的路径:$  cd home/

运行$ sudo sh cuda_7.5.18_linux.run
按照提示一步步操作

注:a、一定要按照提示)输入相应字符,例如有的需要输入accept,有的需要输入yes;

       b、遇到提示是否安装opengl ,选择no(如果你的电脑跟我一样是双显,且主显是非nvidia的gpu在工作需要选择no,否则可以yes),其他都选择yes或者默认即可。(如果您的电脑是双显卡且在这一步选择了yes,那么你极有可能安装完cuda之后,重启图形化界面后遇到登录界面循环问题:输入密码后又跳回密码输入界面。 

这是因为你的电脑是双显,而且用来显示的那块gpu不是nvidia,则opengl libraries就不应该安装,否则你正在使用的那块gpu(非nvidia的gpu)的opengl libraries会被覆盖,然后gui就无法工作了。)

安装成功后,会显示installed,否则会显示failed。

5) 输入 $ sudo service lightdm start 重新启动图形化界面。

同时按住alt + ctrl +f7,返回到图形化登录界面,输入密码登录。
如果能够成功登录,则表示不会遇到循环登录的问题,基本说明cuda的安装成功了。

如果你遇到了重复登陆情况,不用急着重装系统,官方教程上有提及,原因上一步的注中有提及,在安装opengl时你可能不注意选择了yes,请卸载cuda,然后重装。
卸载:由于登陆进入不到图形用户界面(gui),但我们可以进入到文本用户界面(tui)

在登陆界面状态下,按ctrl + alt + f1,进入tui
执行

$ sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
$ sudo /usr/bin/nvidia-uninstall

然后重启

$ sudo reboot

重新安装.run   再次安装时请一定留意,在提示是否安装opengl时,你的是双显卡应该选则n。

6) 重启电脑,检查device node verification。

执行

$ ls /dev/nvidia*

可能出现a、b两种结果,请对号入座。

a、若结果显示

/dev/nvidia0      /dev/nvidiactl      /dev/nvidia-uvm

或显示出类似的信息,应该有三个(包含一个类似/dev/nvidia-nvm的),则安装成功。

b、大多数结果可能会是这样

ls: cannot access/dev/nvidia*: no such file or directory

或是这样的,只出现

/dev/nvidia0     /dev/nvidiactl

a中的一个或两个,但没有/dev/nvidia-num,即文件显示不全。

不用着急也不用急着重装系统(我在安装时就是这种情况),官方指导中有详细的解决方案,但是我的方法和官方稍微有些出入。

首先要添加一个启动脚本(添加启动脚本的方法大致有两种,我采用最直接的方法,另一种可以先创建一个文件然后通过mv的方式移动到启动文件夹下,可自行百度)
执行

$ sudo vi /etc/rc.local

如果你是第一次打开这个文件,它应该是空的(除了一行又一行的#注释项外)。这文件的第一行是

#!/bin/sh -e

把-e去掉(这步很重要,否则它不会加载这文本的内容)
然后把下列内容除了#!/bin/bash外复制到其中,(before exit 0 )保存退出。

#!/bin/bash

/sbin/modprobe nvidia

if [ "$?" -eq 0 ]; then
# count the number of nvidia controllers found.
nvdevs=`lspci | grep -i nvidia`
n3d=`echo "$nvdevs" | grep "3d controller" | wc -l`
nvga=`echo "$nvdevs" | grep "vga compatible controller" | wc -l`

n=`expr $n3d + $nvga - 1`
for i in `seq 0 $n`; do
mknod -m 666 /dev/nvidia$i c 195 $i
done

mknod -m 666 /dev/nvidiactl c 195 255

else
exit 1
fi

/sbin/modprobe nvidia-uvm

if [ "$?" -eq 0 ]; then
# find out the major device number used by the nvidia-uvm driver
d=`grep nvidia-uvm /proc/devices | awk '{print $1}'`

mknod -m 666 /dev/nvidia-uvm c $d 0
else
exit 1
fi

下次重启时,你应该能直接看到/dev目录下的三个nvidia的文件
输入:$ ls /dev/nvidia*
结果显示:/dev/nvidia0       /dev/nvidiactl       /dev/nvidia-uvm

成功!

7) 设置环境变量。

终端中输入 $ sudo gedit /etc/profile
在打开的文件末尾,添加以下两行。

64位系统:

export path=/usr/local/cuda-9.0/bin${path:+:${path}}
export ld_library_path=/usr/local/cuda-9.0/lib64\
                          ${ld_library_path:+:${ld_library_path}}

32位系统:

export path=/usr/local/cuda-9.0/bin${path:+:${path}}

export ld_library_path=/usr/local/cuda-9.0/lib\
                          ${ld_library_path:+:${ld_library_path}}

保存文件,并重启。因为source /etc/profile是临时生效,重启电脑才是永久生效。

这里有点与官方安装文档稍有不同,需要说明:
官方文档里说只需在终端中运行上述两条export语句即可,但如果不将它们不写入/etc/profile文件的话,这样的环境变量在你退出终端后就消失了,不起作用了,所以写入才是永久的做法。

8) 重启电脑,检查上述的环境变量是否设置成功。

a、 验证驱动版本

敲入

$ cat /proc/driver/nvidia/version

结果显示

nvrm version: nvidia unix x86_64 kernel module 384.81 sat sep 2 02:43:11 pdt 2017
gcc version: gcc version 5.4.0 20160609 (ubuntu 5.4.0-6ubuntu1~16.04.5)

b、 验证cuda toolkit

敲入

$ nvcc -v       会输出cuda的版本信息

 如果是这样的:

the program 'nvcc' is currently not installed. you can install it by typing:
sudo apt-get install nvidia-cuda-toolkit

可能是环境配置没有成功,请重复上述步骤7)。

五、 尝试编译cuda提供的例子

1)打开终端输入:$   cd /home/xxx/nvidia_cuda-9.0_samples        其中xxx是你自己的用户名,通过命令cd进入nvidia_cuda-9.0_samples目录。

然后终端输入:$ make

系统就会自动进入到编译过程,整个过程大概需要十几到二十分钟,请耐心等待。如果出现错误的话,系统会立即报错停止。

第一次运行时可能会报错,提示的错误信息可能会是系统中没有gcc,

解决办法就是通过命令重新安装gcc就行,在终端输入:$ sudo apt-get install gcc 安装完gcc后, 再make就正常了。

如果编译成功,最后会显示finished building cuda samples,如下图所示。

Ubuntu上如何安装CUDA

2)运行编译生成的二进制文件。
编译后的二进制文件 默认存放在nvidia_cuda-9.0_samples/bin中。
接着在上一个终端中输入 :$  cd /home/lxxx/nvidia_cuda-9.0_samples/bin/x86_64/linux/release       其中xxx是你自己的用户名
然后在终端输入 :$ ./devicequery  

结果如下图所示:看到类似如下图片中的显示,则代表cuda安装且配置成功,其中 result = pass代表成功,若失败 result = fail

Ubuntu上如何安装CUDA

3)最后再检查一下系统和cuda-capable device的连接情况
终端输入 : $ ./bandwidthtest
看到类似如下图片中的显示,则代表成功

Ubuntu上如何安装CUDA

“Ubuntu上如何安装CUDA”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注亿速云网站,小编将为大家输出更多高质量的实用文章!

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI