ubuntu 20.04上docker 使用gpu

要在Docker容器中使用GPU，你需要确保系统上已经安装了正确的NVIDIA驱动程序，并且安装了NVIDIA Container Toolkit。以下是详细的步骤：

1. 安装NVIDIA驱动程序

确保你的系统上已经安装了适当版本的NVIDIA驱动程序。你可以通过运行以下命令来检查驱动程序是否正确安装：

nvidia-smi

如果你看到GPU信息，那么驱动程序已经正确安装。

2. 安装Docker

如果你还没有安装Docker，可以按照以下步骤安装：

# 更新包索引
sudo apt-get update

# 安装Docker依赖包
sudo apt-get install -y apt-transport-https ca-certificates curl software-properties-common

# 添加Docker官方GPG密钥