发布于 

CUDA编程笔记001. 安装CUDA

操作环境：Dell笔记本

  • Ubuntu 22.04 x86_64
  • Nvidia Geforce GTX 1050Ti

安装指导：https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html 3.10章节

1、检查显卡硬件

$ lspci | grep -i nvidia
01:00.0 VGA compatible controller: NVIDIA Corporation GP107M [GeForce GTX 1050 Ti Mobile] (rev a1)
01:00.1 Audio device: NVIDIA Corporation GP107GL High Definition Audio Controller (rev a1)

上述信息表示GPU驱动已安装完毕。

2、安装内核头文件和开发库

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

检查是否安装成功：

pkg-query -s linux-headers-$(uname -r)

3、安装CUDA

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda

4、设置CUDA环境变量

在~/.bashrc或~/.zshrc中追加以下命令：

export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}

5、重启机器。

sudo reboot

6、验证安装成功

  • 查看驱动版本：
cat /proc/driver/nvidia/version
  • 查看GPU
nvidia-smi
$ git clone https://github.com/NVIDIA/cuda-samples.git
$ cd cuda-samples
$ make
$ ./bin/x86_64/linux/release/asyncAPI
[./bin/x86_64/linux/release/asyncAPI] - Starting...
GPU Device 0: "Pascal" with compute capability 6.1

CUDA device [NVIDIA GeForce GTX 1050 Ti]
time spent executing by the GPU: 13.18
time spent by CPU in CUDA calls: 6.04
CPU executed 29511 iterations while waiting for GPU to finish

运行成功，表示CUDA正常工作。

以上便是CUDA的安装流程。

许可协议

本文采用 署名-非商业性使用-相同方式共享 4.0 国际 许可协议，转载请注明出处。