要在Docker容器中使用GPU，你需要確保系統(tǒng)上已經(jīng)安裝了正確的NVIDIA驅(qū)動程序，并且安裝了NVIDIA Container Toolkit。以下是詳細的步驟：

1. 安裝NVIDIA驅(qū)動程序

確保你的系統(tǒng)上已經(jīng)安裝了適當(dāng)版本的NVIDIA驅(qū)動程序。你可以通過運行以下命令來檢查驅(qū)動程序是否正確安裝：

nvidia-smi

如果你看到GPU信息，那么驅(qū)動程序已經(jīng)正確安裝。

2. 安裝Docker

如果你還沒有安裝Docker，可以按照以下步驟安裝：

# 更新包索引
sudo apt-get update# 安裝Docker依賴包
sudo apt-get install -y apt-transport-https ca-certificates curl software-properties-common# 添加Docker官方GPG密鑰