如何将NVIDIA和CUDA驱动程序安装到Ubuntu中?


Answers:


52

Ubuntu 18.04,CUDA 10.1,libcudnn 7.5.1和NVIDIA 418.67驱动程序:

注意2019年6月23日: CUDA 10.0或10.1版本的最新更新(随NVIDIA 418.67驱动程序一起安装)不再包含32位库,这将导致Steam和大多数游戏无法正常工作。的版本libnvidia-gl-418:i386仅安装418.56版本,而418.67驱动程序将不起作用。希望NVIDIA会尽快发布更新。我已在此答案的底部添加了run file install有关如何仅下载CUDA安装程序的运行文件的信息,然后可以使用所需的任何驱动程序。 运行文件的大小为2.3GB,因此下载可能需要一些时间。

注意: CUDA 9.x无法通过NVIDIA的ubuntu1804存储库获得。 但是我确实在 https://askubuntu.com/a/1086993/231142上为CUDA 9.2编写了答案

这些说明用于通过存储库而非安装来安装CUDA .deb

您可以将以下几行复制并粘贴到终端窗口中。按Ctrl+ Alt+ T打开终端窗口。

删除所有可能设置的CUDA PPA,并删除nvidia-cuda-toolkit已安装的CUDA PPA :

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove --autoremove nvidia-cuda-toolkit

建议在安装新驱动程序之前也删除所有NVIDIA驱动程序:

sudo apt remove --autoremove nvidia-*

然后更新系统:

sudo apt update

最近,我刚刚发现CUDA安装可与一起使用,graphics-drivers ppa因此如果您没有添加,请立即添加:

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

安装密钥:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

添加仓库:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda_learn.list'

再次更新系统:

sudo apt update

安装CUDA 10.1:

sudo apt install cuda-10-1

它应该随其一起安装NVIDIA 418.40驱动程序,因为这些都是仓库中列出的内容。请参阅:http : //developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/

安装libcudnn7 7.5.1:

sudo apt install libcudnn7

~/.profile将以下行添加到CUDA 10.1的文件中

# set PATH for cuda 10.1 installation
if [ -d "/usr/local/cuda-10.1/bin/" ]; then
    export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

重新启动计算机,并在完成重新启动后检查设置:

使用以下命令检查NVIDIA Cuda Compiler nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Apr_24_19:10:27_PDT_2019
Cuda compilation tools, release 10.1, V10.1.168

检查libcudnn版本/sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn

terrance@terrance-ubuntu:~$ /sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn
    libcudnn.so.7 -> libcudnn.so.7.5.1

使用以下命令检查NVIDIA驱动程序nvidia-smi

terrance@terrance-ubuntu:~$ nvidia-smi 
Sat Jun  1 09:38:07 2019       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.67       Driver Version: 418.67       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 750 Ti  On   | 00000000:02:00.0  On |                  N/A |
| 40%   38C    P0     2W /  38W |    116MiB /  2000MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      2216      G   /usr/lib/xorg/Xorg                           110MiB |
|    0      2542      G   compton                                        1MiB |
+-----------------------------------------------------------------------------+

.run 文件安装

通过使用,sudo add-apt-repository ppa:graphics-drivers/ppa您可以安装430.26最新的驱动程序或任何您喜欢的驱动程序。

接下来,通过以下步骤安装libcudnn7:

添加仓库:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda_learn.list'

安装密钥:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

更新系统:

sudo apt update

安装libcudnn7.5.1:

sudo apt install libcudnn7

现在cuda_10.1.105_418.39_linux.runhttps://developer.nvidia.com/cuda-10.1-download-archive-base?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal下载

然后运行安装程序:

sudo sh cuda_10.1.105_418.39_linux.run

输入accept,然后在此屏幕上按Enter:

┌──────────────────────────────────────────────────────────────────────────────┐
│  End User License Agreement                                                  │
│  --------------------------                                                  │
│                                                                              │
│                                                                              │
│  Preface                                                                     │
│  -------                                                                     │
│                                                                              │
│  The Software License Agreement in Chapter 1 and the Supplement              │
│  in Chapter 2 contain license terms and conditions that govern               │
│  the use of NVIDIA software. By accepting this agreement, you                │
│  agree to comply with all the terms and conditions applicable                │
│  to the product(s) included herein.                                          │
│                                                                              │
│                                                                              │
│  NVIDIA Driver                                                               │
│                                                                              │
│                                                                              │
│  Description                                                                 │
│                                                                              │
│  This package contains the operating system driver and                       │
│──────────────────────────────────────────────────────────────────────────────│
│ Do you accept the above EULA? (accept/decline/quit):                         │
│ accept                                                                       

取消选择驱动程序,然后使用箭头键和空格键选择安装以移动并选择或取消选择:

┌──────────────────────────────────────────────────────────────────────────────┐
│ CUDA Installer                                                               │
│ - [ ] Driver                                                                 │
│      [ ] 418.39                                                              │
│ + [X] CUDA Toolkit 10.1                                                      │
│   [X] CUDA Samples 10.1                                                      │
│   [X] CUDA Demo Suite 10.1                                                   │
│   [X] CUDA Documentation 10.1                                                │
│   Install                                                                    │
│   Options                                                                    │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│                                                                              │
│ Up/Down: Move | Left/Right: Expand | 'Enter': Select | 'A': Advanced options │

等待安装完成,它可能会在安装过程中提示错误,但请放心。

~/.profile将以下行添加到CUDA 10.1的文件中

# set PATH for cuda 10.1 installation
if [ -d "/usr/local/cuda-10.1/bin/" ]; then
    export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

重新引导系统以使更改生效。


1
请注意,从2018-11开始,张量流需要cuda 0.9。
mathtick

1
nvidia-driver-410的安装失败,因为它依赖于xserver-xorg-video-nvidia-410,并且无法安装,因为它依赖于一堆不可用的
lbnvidia

1
@fccoelho我最近发现该安装实际上与结合使用效果最好graphics-drivers ppa,因此我在这里更新了我的答案以包含该PPA,然后它应该找到并安装所有依赖项。
Terrance

1
删除以前的工具包和驱动程序后,应该执行“ sudo apt autoremove”步骤。否则,由于依赖关系不会自动更新,安装将失败。
Pieter-Jan Busschaert

1
附加说明:NVidia的10.1 CUDA .deb软件包中包含的NVidia驱动程序不包括32位库。安装这种方式后,Steam(和大多数游戏)将不再起作用。另请参见:github.com/ValveSoftware/steam-for-linux/issues/5778
Pieter-Jan Busschaert

8

Ubuntu 16.04,CUDA 9.2和NVIDIA 396驱动程序:

注意:此处的NVIDIA存储库已决定推送410驱动程序。我将进行一些测试,以查看是否可以为要安装的驱动程序进行设置。

这些说明用于通过存储库而非安装来安装CUDA .deb

您可以将以下几行复制并粘贴到终端窗口中。按Ctrl+ Alt+ T打开终端窗口。

删除所有可能设置的CUDA PPA,并删除nvidia-cuda-toolkit已安装的CUDA PPA :

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove nvidia-cuda-toolkit

建议在安装新驱动程序之前也删除所有NVIDIA驱动程序:

sudo apt remove nvidia-*

然后更新系统:

sudo apt update

安装密钥:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

添加仓库:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

已添加新仓库的更新:

sudo apt update

安装CUDA 9.2:

sudo apt install cuda-9-2

它应该随其一起安装nvidia-396驱动程序,因为这些都在存储库中列出。请参阅:http : //developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/

验证是否已安装CUDA 9.2:

~$ ls /usr/local/cuda-9.2/
bin     include    libnvvp           nvml    samples  targets
doc     lib64      LICENSE           nvvm    share    tools
extras  libnsight  nsightee_plugins  README  src      version.txt

现在,~/.profilePATH和添加以下内容LD_LIBRARY。您可以使用以下命令gedit ~/.profile进行编辑:

# set PATH for cuda 9.2 installation
if [ -d "/usr/local/cuda-9.2/bin/" ]; then
    export PATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

重新启动系统。

sudo reboot

系统启动后,可以通过键入以下内容来验证安装:

nvcc -V

您应该看到类似以下内容:

~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

并且您应该看到396.xx安装了驱动程序:

~$ nvidia-smi
Thu May 17 07:38:54 2018        
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 396.44                 Driver Version: 396.44                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 760     Off  | 00000000:02:00.0 N/A |                  N/A |
| 49%   53C    P0    N/A /  N/A |    187MiB /  1999MiB |     N/A      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0                    Not Supported                                       |
+-----------------------------------------------------------------------------+


3

Ubuntu 14.04,CUDA 10.0和NVIDIA 410驱动程序:

请注意,我没有安装Ubuntu 14.04,因此无法验证这些步骤是否有效。但是根据NVIDIA的http://developer.download.nvidia.com/compute/cuda/repos/,它应该与我在其他两个答案中列出的步骤相同。

这些说明用于通过存储库而非安装来安装CUDA .deb

您可以将以下几行复制并粘贴到终端窗口中。按Ctrl+ Alt+ T打开终端窗口。

删除所有可能设置的CUDA PPA,并删除nvidia-cuda-toolkit已安装的CUDA PPA :

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove nvidia-cuda-toolkit

建议在安装新驱动程序之前也删除所有NVIDIA驱动程序:

sudo apt remove nvidia-*

然后更新系统:

sudo apt update

安装密钥:

sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/7fa2af80.pub

添加仓库:

sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

再次更新系统:

sudo apt update

安装CUDA 10.0。

sudo apt install cuda-10-0

它应该随其一起安装nvidia-410驱动程序,因为这些都在存储库中列出。请参阅:http : //developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/

~/.profile将以下行添加到CUDA 10.0的文件中

# set PATH for cuda 10.0 installation
if [ -d "/usr/local/cuda-10.0/bin/" ]; then
    export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

重新启动计算机,并在完成重新启动后检查设置:

使用以下命令检查NVIDIA Cuda Compiler nvcc --version

使用以下命令检查NVIDIA驱动程序nvidia-smi


0

不要犯错。永久下载.deb文件。我下载了.run文件并安装了它。但是,一旦我尝试安装tensorrt,事实证明只有在您的系统使用nvidia驱动程序的.deb文件安装的情况下才能安装。

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.