安装前提:
1、显卡驱动安装。先确保你已经安装了NVIDIA的显卡驱动。你可以通过运行nvidia-smi命令来检查驱动是否已经安装。如果驱动已经安装,这个命令会显示你的GPU的详细信息。
2、安装docker。确保docker已经正确安装。
安装NVIDIA-docker
你可以运行以下命令来安装:
# 添加NVIDIA的GPG key
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
# 添加NVIDIA-docker的repository
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-docker2
重新启动docker:
sudo systemctl restart docker
现在,你应该已经成功安装了NVIDIA-docker,你可以通过运行docker run --gpus all nvidia/cuda:10.0-base nvidia-smi来测试是否可以在Docker容器中使用你的GPU。