Linux下的CUDA安装指南 (在linux中cuda怎么安装教程)
随着深度学习和领域的迅速发展和普及,GPU计算技术越来越被广泛使用。CUDA是Nvidia公司针对其GPU推出的并行计算平台和编程模型,可以使我们利用GPU的强大计算能力加速程序的运算速度。但是,对于初学者来说,在Linux下安装CUDA还是有些棘手的。为此,本文将提供一份详细易懂的Linux下CUDA安装指南。
一、准备工作
在进行CUDA安装前,需要进行一些前期的准备工作。
1.确认你的计算机支持Nvidia GPU。可以使用以下命令来检测:
$ lspci | grep -i nvidia
2.确认你的GPU型号。可以使用以下命令来检测:
$ nvidia-i
3.确认你的Linux系统是否支持CUDA。可以在Nvidia的官网上查看。
4.下载CUDA的安装文件。可以在Nvidia的官网上下载。
二、安装CUDA
下面将分步骤介绍如何在Linux下安装CUDA。
1.卸载现有的Nvidia驱动程序。可以使用以下命令:
$ sudo apt-get remove –purge nvidia*
2.卸载Nvidia相关组件。可以使用以下命令:
$ sudo apt-get autoremove
3.禁用Nouveau驱动程序。可以使用以下命令:
$ sudo vim /etc/modprobe.d/blacklist-nouveau.conf
在文件中加入以下内容:
blacklist nouveau
options nouveau modeset=0
4.安装gcc和make工具。可以使用以下命令:
$ sudo apt-get install gcc make
5.重启计算机。为了确保之前的操作生效,需要重启计算机。
6.进入命令行模式。在电脑开机时,按下Ctrl+Alt+F1进入命令行模式。
7.禁用X服务。可以使用以下命令:
$ sudo service lightdm stop
8.运行安装程序。可以使用以下命令:
$ sudo sh cuda_*.run
在安装过程中会提示你如何进行各种配置和选择。按照提示进行选择即可。
9.重启计算机。为了确保CUDA的安装生效,还需要再次重启计算机。
三、测试CUDA
在安装CUDA之后,可以通过以下步骤来测试CUDA是否正确安装。
1.打开Nvidia的示例程序。可以使用以下命令:
$ cd /usr/local/cuda/samples
$ make
2.执行CUDA示例程序。可以使用以下命令:
$ sudo ./bin/x86_64/linux/release/deviceQuery
执行完之后,如果看到以下提示,则说明CUDA已经安装成功。
CUDA Device Query (Runtime API) version (CUDART static linking)
Detected 1 CUDA Capable device(s)
Device 0: “GeForce GTX 1070”
CUDA Driver Version / Runtime Version 9.2 / 9.2
CUDA Capability Major/Minor version number: 6.1
Total amount of global memory: 8119 MBytes (8517871616 bytes)
(15) Multiprocessors, (128) CUDA Cores/MP: 1920 CUDA Cores
GPU Max Clock rate: 1683 MHz (1.68 GHz)
Memory Clock rate: 4004 Mhz
Memory Bus Width: 256-bit
L2 Cache Size: 2023152 bytes
Maximum Texture Dimension Size (x,y,z) 1D=(131072), 2D=(131072, 65536), 3D=(16384, 16384, 16384)
Maximum Layered 1D Texture Size, (num) layers 1D=(32768), 2023 layers
Maximum Layered 2D Texture Size, (num) layers 2D=(32768, 32768), 2023 layers
Total amount of constant memory: 65536 bytes
Total amount of shared memory per block: 49152 bytes
Total number of registers avlable per block: 65536
Warp size: 32
Maximum number of threads per multiprocessor: 2023
Maximum number of threads per block: 1024
Max dimension size of a thread block (x,y,z): (1024, 1024, 64)
Max dimension size of a grid size (x,y,z): (2147483647, 65535, 65535)
Maximum memory pitch: 2147483647 bytes
Texture alignment: 512 bytes
Concurrent copy and kernel execution: Yes with 2 copy engine(s)
Run time limit on kernels: No
Integrated GPU sharing Host Memory: Yes
Support host page-locked memory mapping: Yes
Alignment requirement for Surfaces: Yes
Device has ECC support: Disabled
CUDA Device Driver Mode (TCC or WDDM): WDDM (Windows Display Driver Model)
Device supports Unified Addressing (UVA): Yes
Device supports Compute Preemption: Yes
Supports Cooperative Kernel Launch: Yes
Supports MultiDevice Co-op Kernel Launch: Yes
Device PCI Domn ID / Bus ID / location ID: 0 / 1 / 0
Compute Mode:
Default (multiple host threads can use this device simultaneously)
deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 9.2, CUDA Runtime Version = 9.2, NumDevs = 1, Device0 = GeForce GTX 1070
Result = PASS
四、
在本文中,我们介绍了如何在Linux下安装CUDA。虽然在安装过程中需要进行一些较为繁琐的配置,但只要按照本文的步骤进行操作,就能够顺利地完成安装。希望本文可以对初学者在Linux下使用CUDA有所帮助。