Tutorial Lengkap Install NVIDIA Docker di Ubuntu

1. Prasyarat Sistem

Pastikan sistem memenuhi syarat berikut:

GPU NVIDIA terpasang

Akses sudo

Cek GPU:

lspci | grep -i nvidia

Cek driver:

nvidia-smi

Jika nvidia-smi gagal, install driver terlebih dahulu:

sudo ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
sudo reboot

2. Install Docker Engine

Install dependency

sudo apt update
sudo apt install -y ca-certificates curl gnupg lsb-release

Tambahkan Docker GPG key

sudo mkdir -p /etc/apt/keyrings
curl -fsSL https://download.docker.com/linux/ubuntu/gpg \
 | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg

Tambahkan Docker repository

echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] \
https://download.docker.com/linux/ubuntu $(lsbrelease -cs) stable" \

| sudo tee /etc/apt/sources.list.d/docker.list > /dev/null

sudo apt update
sudo apt install -y docker-ce docker-ce-cli containerd.io

sudo usermod -aG docker $USER
newgrp docker

docker run hello-world

curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
 | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg

curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
 | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
 | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

sudo apt update
sudo apt install -y nvidia-container-toolkit

sudo nvidia-ctk runtime configure --runtime=docker

sudo systemctl restart docker

docker run --rm --gpus all \
  nvidia/cuda:12.4.1-base-ubuntu22.04 nvidia-smi

docker run --rm -it --gpus all \
  pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime \
  python -c "import torch; print(torch.cuda.isavailable())"

Output yang benar:

True

7. Verifikasi Runtime Docker

docker info | grep -i runtime

Pastikan nvidia muncul dalam daftar runtime.

8. Troubleshooting Umum

Error: could not select device driver "nvidia"

Solusi:

sudo systemctl restart docker
sudo reboot

CUDA tidak terdeteksi di container

Cek library CUDA di host:

ls /usr/lib/x86_64-linux-gnu/libcuda.so

NVIDIA Docker sudah termasuk dalam JetPack

• Jangan install NVIDIA Container Toolkit manual

9. Alur Instalasi (Ringkas)

NVIDIA Driver (host)
      ↓
Docker Engine
      ↓
NVIDIA Container Toolkit
      ↓
docker run --gpus all
      ↓
CUDA tersedia di container