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