📜  安装 docker nvidia 2 - Shell-Bash (1)

📅  最后修改于: 2023-12-03 15:25:05.465000             🧑  作者: Mango

介绍

Docker是一个开源的容器化平台,可以帮助开发人员更方便地构建,打包和分发应用程序。与传统的虚拟机不同,Docker利用操作系统级虚拟化技术,使得容器化应用程序在部署和运行时更加轻量化和高效。此外,Docker可以与NVIDIA GPU一起使用,以更好地支持容器化深度学习应用程序。本文将介绍如何在Linux系统上安装Docker和NVIDIA Docker组件,以便进行深度学习容器的构建和管理。

安装Docker

Docker的安装可以通过各种方式进行,包括二进制文件包,官方的存储库等等。以下是在Ubuntu系统上使用官方存储库安装Docker的步骤:

  1. 移除旧版本Docker:
sudo apt-get remove docker docker-engine docker.io containerd runc
  1. 安装Docker的依赖库:
sudo apt-get update
sudo apt-get install apt-transport-https ca-certificates curl gnupg lsb-release
  1. 添加Docker的官方GPG密钥:
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg
  1. 添加Docker的官方APT存储库:
echo "deb [arch=amd64 signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
  1. 安装Docker:
sudo apt-get update
sudo apt-get install docker-ce docker-ce-cli containerd.io
  1. 确认Docker已成功安装:
sudo docker run hello-world

安装NVIDIA Docker 2

要在Docker容器中使用GPU,需要安装NVIDIA Docker 2。以下是在Ubuntu系统上安装NVIDIA Docker 2的步骤:

  1. 卸载旧版NVIDIA Docker:
sudo apt-get remove nvidia-docker
  1. 添加NVIDIA Docker 2的APT存储库:
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
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
  1. 安装NVIDIA Docker 2:
sudo apt-get update
sudo apt-get install -y nvidia-docker2
  1. 确认NVIDIA Docker 2已成功安装:
sudo docker run --gpus all nvidia/cuda:10.0-base nvidia-smi

以上为在Ubuntu系统上安装Docker和NVIDIA Docker 2的步骤。请根据您的实际情况进行调整。安装完成后,您可以使用Docker来构建和管理容器化的深度学习应用程序,而NVIDIA Docker 2可以帮助您更好地支持GPU加速和容器化。