Chuyển đến nội dung chính
Phiên bản: 2.0.0

Cài đặt CUDA trên Linux

Nếu bạn đang sử dụng phiên bản máy ảo hỗ trợ Nvidia từ đám mây công cộng, có lẽ bạn nên sử dụng hình ảnh máy ảo do đám mây cung cấp. Nó thường có các phiên bản chính xác của trình điều khiển Nvidia và bộ công cụ CUDA đã được cài đặt. Đọc tiếp nếu bạn cần cài đặt trình điều khiển Nvidia và bộ công cụ CUDA trên máy của riêng bạn.

Ubuntu 22.04​

1 Cài đặt trình điều khiển Nvidia.​

Xây dựng lại cấu hình grub:

sudo apt-get install -y gcc make linux-headers-$(uname -r)
cat << EOF | sudo tee --append /etc/modprobe.d/blacklist.conf
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
EOF
sudo sed -i 's/GRUB_CMDLINE_LINUX=""/GRUB_CMDLINE_LINUX="rdblacklist=nouveau"/' /etc/default/grub
sudo update-grub

Tải xuống và cài đặt trình điều khiển Nvidia

WGET https://storage.googleapis.com/nvidia-drivers-us-public/GRID/vGPU16.1/NVIDIA-Linux-x86_64-535.104.05-grid.run
sudo sh NVIDIA-Linux-x86_64-535.104.05-grid.run

Xác nhận trình điều khiển đã được cài đặt thành công

NVIDIA-SMI -Q | đầu

==============NHẬT KÝ NVSMI==============

Dấu thời gian : Fri Oct 27 21:54:05 2023
Phiên bản trình điều khiển : 535.104.05
Phiên bản CUDA : 12.2

GPU kèm theo : 1
GPU 00000000:00:1E.0
    Tên sản phẩm : NVIDIA A10G

Vô hiệu hóa GSP và khởi động lại.

sudo chạm /etc/modprobe.d/nvidia.conf
echo "Tùy chọn Nvidia NVreg_EnableGpuFirmware = 0" | sudo tee --nối thêm /etc/modprobe.d/nvidia.conf
sudo khởi động lại

2 Cài đặt bộ công cụ CUDA.​

WGET https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run
sudo sh cuda_12.2.2_535.104.05_linux.run --silent --override --toolkit --sample --toolkitpath = / usr / local / cuda-12 --samplespath = / usr / local / cuda --no-opengl-libs

Xác nhận rằng CUDA đã được cài đặt.

/usr/local/cuda/bin/nvcc --version

nvcc: Trình điều khiển trình biên dịch NVIDIA (R) Cuda
Bản quyền (c) 2005-2023 NVIDIA Corporation
Được xây dựng trên Tue_Aug_15_22:02:13_PDT_2023
Công cụ biên dịch Cuda, phát hành 12.2, V12.2.140
Bản dựng cuda_12.2.r12.2/compiler.33191640_0

Sau đó, sử dụng hai lệnh sau để thiết lập đường dẫn môi trường. Có lẽ bạn nên thêm hai dòng này vào ~/.bashrc và ~/.bash_profile (hoặc ~/.zshrc và ~/.Hồ sơ) để các thiết bị đầu cuối mới và thông tin đăng nhập trong tương lai vẫn có thể tìm thấy các tệp thư viện CUDA này.

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}
export PATH=/usr/local/cuda/bin:${PATH}

Tài nguyên khác​

Dưới đây là các tập lệnh khác có thể giúp bạn trong trường hợp bạn gặp khó khăn.