如何在Ubuntu 18.04上安装CUDA

时间:2020-03-05 15:32:25  来源:igfitidea点击:

CUDA旨在通过利用系统上的图形处理单元(GPU)的功率来实现计算性能的显着增加。
在本文中,我会告诉我们如何在Ubuntu 18.04上安装CUDA。
CUDA是由NVIDIA开发的并行编程模型和计算平台。

我们可以使用以下方法之一在Ubuntu 18.04上安装CUDA:

  • 从独立分配的包(Runfile包)
  • 从分发特定的包(RPM和DEB包)

CUDA预安装检查

以下是在Ubuntu系统上安装CUDA Toolkit和Driver之前需要执行的许多检查。

验证系统是否有CUDA的GPU

我们需要验证GPU可以使用CUDA,请运行以下命令查看:

$lspci | grep -i nvidia
01:00.0 3D controller: NVIDIA Corporation GM206M [GeForce GTX 965M] (rev a1)

如果我们刚刚安装了驱动程序,则可能需要手动更新PCI数据库以获取上面的命令以返回有效输出。

# update-pciids

验证系统是否安装了GCC

对于使用CUDA的开发,我们需要确保安装了GCC。
我们可以检查它是否已使用命令安装:

$gcc --version

如果未安装,请使用apt-得到如下安装:

$sudo apt install gcc-6 g++-6

验证系统是否安装了正确的内核标题和开发包。

CUDA驱动程序要求在驱动程序安装时安装内核版本的内核和开发包,同时只要驱动程序重建驱动程序。
我们可以使用以下内容安装内核标题和开发工具:

$sudo apt-get install linux-headers-$(uname -r)

安装nvidia驱动程序

CUDA需要在机器上安装NVIDIA驱动程序。
使用命令将其安装在Ubuntu 18.04上:

$sudo apt install nvidia-384

安装后,我们可以继续安装NVIDIA CUDA Toolkit。

下载nvidia cuda工具包

根据安装方法,我们需要下载等效包。
我更喜欢从Ubuntu 18.04上的Runfile安装CUDA,因为它很难遇到依赖性问题。

截至本文的撰写,最新发布的CUDA是V9.2.
由于包大小高于1GB,我将使用wget命令下载它,以便如果连接破坏,我可以轻松恢复。
CUDA Toolkit包含CUDA驱动程序和工具所需的创建,构建和运行CUDA应用程序以及库,标题文件,CUDA示例源代码和其他资源。

$cd Dowloads
$wget -c https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda_9.2.88_396.26_linux

一旦软件包在本地下载,使其可执行并安装它。

# chmod +x cuda_9.2.88_396.26_linux.run
# ./cuda_9.2.88_396.26_linux.run --verbose --silent --toolkit --override

我们应该在完整的安装时获得类似于下面的输出。

===========
= Summary =
===========
Toolkit: Installed in /usr/local/cuda-9.2
Samples: Not Selected
Please make sure that
 - PATH includes /usr/local/cuda-9.2/bin
 - LD_LIBRARY_PATH includes /usr/local/cuda-9.2/lib64, or, add /usr/local/cuda-9.2/lib64 to /etc/ld.so.conf and run ldconfig as root

修改.bashrc文件以在其路径中包含CUDA BIN:

export PATH="$PATH:/usr/local/cuda-9.2/bin"

确保CUDA库路径存在。

# echo "/usr/local/cuda-9.2/lib64" >> /etc/ld.so.conf
# ldconfig

检查https://developer.nvidia.com/cuda-downloads for可用修补程序,然后下载.run文件然后安装它。

# wget https://developer.nvidia.com/compute/cuda/9.2/Prod/patches/1/cuda_9.2.88.1_linux
# chmod +x cuda_9.2.88.1_linux.run
# ./cuda_9.2.88.1_linux.run --silent --accept-eula
Welcome to the CUDA Patcher.
Installation complete!
Installation directory: /usr/local/cuda-9.2

在CUDA BIN文件夹中为GCC6创建符号链接:

# ln -s /usr/bin/gcc-6 /usr/local/cuda-9.2/bin/gcc
# ln -s /usr/bin/g++-6 /usr/local/cuda-9.2/bin/g++