Avatar
Ba
Hướng dẫn cài CUDA trên Ubuntu

Cài đặt CUDA trên Ubuntu đảm bảo thành công

Để cài CUDA trên Ubuntu luôn là một thách thức. Sau đây mình sẽ trình bày một số dòng lệnh cơ bản để bạn có thể cài được Ubuntu hiệu quả.

Cài đặt NVIDIA drivers


Cập nhật và nâng cấp hệ thống

sudo apt update && sudo apt upgrade

Gỡ cài đặt NVIDIA trước đó

sudo apt autoremove nvidia* --purge

Kiểm tra thiết bị trên Ubuntu

ubuntu-drivers devices

Nếu bạn chưa chạy được dòng lệnh bên trên ngay, bạn có thể sử dụng dòng lệnh sau:

apt install ubuntu-drivers-common

Cài đặt Ubuntu drivers

sudo ubuntu-drivers autoinstall

Cài đặt NVIDIA drivers Phiên bản đề xuất của tôi là 525, điều chỉnh phiên bản phù hợp với của bạn

sudo apt install nvidia-driver-525

Khởi động lại máy & Kiểm tra

reboot

sau đó chạy dòng lệnh sau để biết rằng driver đã cài đặt thành công

nvidia-smi

Cài đặt CUDA drivers


Cập nhật và nâng cấp hệ thống

sudo apt update && sudo apt upgrade

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

sudo apt install nvidia-cuda-toolkit

Kiểm tra việc cài đặt CUDA

nvcc --version

Đôi khi các bạn làm AI và chạy các docker yêu cầu sử dụng card đồ họa và gặp lỗi như sau:

docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].

Cách fix:

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

Cài đặt cuDNN


Tải tệp .deb của cuDNN Bạn có thể tải tệp cuDNN tại đây. Bạn sẽ cần một tài khoản Nvidia.

Chọn phiên bản cuDNN phù hợp với phiên bản CUDA tương ứng, mà bạn có thể xem khi chạy lệnh sau:

nvcc --version

Cài đặt cuDNN

sudo apt install ./<tên_tệp.deb>
sudo cp /var/cudnn-<tệp>.gpg /usr/share/keyrings/

Phiên bản cuDNN của tôi là 8, điều chỉnh theo phiên bản của bạn:

sudo apt update
sudo apt install libcudnn8
sudo apt install libcudnn8-dev
sudo apt install libcudnn8-samples