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有所帮助。


数据运维技术 » Linux下的CUDA安装指南 (在linux中cuda怎么安装教程)