XYU

Engineering Cybernetics.                    

Install CUDA Toolkit on Ubuntu

26 Nov 2017 » Linux

一、 下载

官网下载最新版本的CUDA Toolkit,在此以Ubuntu 16.04,x86_64的runfile(local)为例。

二、 安装前的配置工作

安装CUDA前,需要先安装NVIDIA的闭源显卡驱动,好在NVIDIA提供的runfile里是包含显卡驱动的,而且在安装CUDA的过程中也会提醒我们安装显卡驱动。在安装显卡驱动之前,我们需要彻底关闭Ubuntu的图形界面保证安装过程和今后的使用不会出现异常。

1、 关闭lightdm

同时按下Ctrl + Alt + F1进入命令行模式,使用如下命令关闭lightdm:

~$ sudo /etc/init.d/lightdm stop

2、 彻底禁用Nouveau

Nouveau是Ubuntu默认使用的开源NVIDIA显卡驱动,在安装NVIDIA的官方驱动之前,需要彻底禁用Nouveau以免两款显卡驱动出现冲突。在/etc/modprobe.d/nvidia-installer-disable-nouveau.conf中添加两行代码:

blacklist nouveau
options nouveau modeset=0

输入以下命令使该配置生效:

~$ sudo blacklist nouveau
~$ sudo options nouveau modeset=0

重新启动计算机。

3、 再次关闭lightdm

重启完成进入图形界面时,再次同时按下Ctrl + Alt + F1进入命令行模式,使用如下命令关闭lightdm:

~$ sudo /etc/init.d/lightdm stop

接下来便可以正式进入到安装过程了。

三、 安装NVIDIA闭源驱动与CUDA

进入runfile所在目录,输入:

~$ sudo sh cuda_9.0.176_384.81_linux.run

在安装过程需要注意以下几点:

  • Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81? yes

  • Do you want to install the OpenGL libraries? no

  • Do you want to run nvidia-xconfig? This will update the system X configuration file so that the NVIDIA X driver is used. The pre-existing X configuration file will be backed up.
    This option should not be used on systems that require a custom X configuration, such as systems with multiple GPU vendors. no

其余选项根据自己需要进行配置即可。
安装完成后重新启动计算机,设置环境变量和动态链接库,在/etc/profile中添加:

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

生效:

~$ source /etc/profile

四、测试安装是否成功

/usr/local/cuda-9.0/samples目录下运行命令make,测试samples是否能够成功编译。

© 2019 XYU.