如何在Ubuntu 18.04上安装CUDA
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++