在 Ubuntu 上安裝 NVIDIA Container Toolkit 的完整流程,讓 Docker 容器可以使用 GPU(如:NVIDIA Jetson、RTX 顯示卡、CUDA 開發等)。
✅ 指令逐步說明:
🔹 1. 加入 NVIDIA GPG 金鑰(供 apt 驗證)
這會下載 GPG 金鑰並儲存為 .gpg
格式到 /usr/share/keyrings
→ 供接下來 apt repository 使用,避免「未簽署的來源」警告。
🔹 2. 加入 NVIDIA APT 套件來源清單(stable 版本)
這段會:
-
下載官方套件來源 list 檔
-
插入 GPG key 使用的
signed-by
參數 -
儲存成
/etc/apt/sources.list.d/nvidia-container-toolkit.list
🔹 3. 啟用 experimental 測試源(某些版本預設註解)
這會移除 #
註解符號,啟用套件清單中名為 experimental
的項目(若有)
🔹 4. 更新 apt 套件清單
🔹 5. 安裝 NVIDIA Container Toolkit 主套件
這會安裝:
-
/usr/bin/nvidia-container-runtime
-
GPU runtime hooks
-
Docker engine GPU 插件
🔹 6. 重新啟動 Docker 服務
→ 確保 docker 套用新的 runtime 設定
→ 接下來 docker run --gpus all
就可以正常啟用 NVIDIA GPU
✅ 檢查是否安裝成功
如果看到 nvidia-smi
顯示 GPU 列表,代表一切設定完成 ✅
📌 延伸建議:加入預設 docker 設定
確認 /etc/docker/daemon.json
包含:
如果你常跑 GPU 容器,這樣可自動使用 NVIDIA runtime。
#restart docker and test docker gpu
$sudo systemctl restart docker
$docker run --rm --gpus all nvidia/cuda:12.2.0-base-ubuntu22.04 nvidia-smi
沒有留言:
張貼留言