要在Docker容器中使用GPU，你需要确保系统上已经安装了正确的NVIDIA驱动程序，并且安装了NVIDIA Container Toolkit。以下是详细的步骤：
1. 安装NVIDIA驱动程序
确保你的系统上已经安装了适当版本的NVIDIA驱动程序。你可以通过运行以下命令来检查驱动程序是否正确安装：
代码语言：javascript复制nvidia-smi如果你看到GPU信息，那么驱动程序已经正确安装。
2. 安装Docker
如果你还没有安装Docker，可以按照以下步骤安装：
代码语言：javascript复制# 更新包索引
sudo apt-get update
# 安装Docker依赖包
sudo apt-get install -y apt-transport-https ca-certificates curl software-properties-common
# 添加Docker官方GPG密钥
