使用 Docker 安装 Animius

为了简化并容器化 Animius, 我们创建了一个 Docker 镜像来承载虚拟环境并分离 Animiuis 的安装流程. Animius 在这个可以与主机共享资源的虚拟环境中运行, 例如使用 GPU (仅限 Linux 系统), 连接互联网等.

在 Linux 系统中, Docker 是最简单的启用 GPU 的方式, 因为用户只需要在主机上安装 英伟达显卡驱动. (无需安装 NVIDIA® CUDA® Toolkit )

Animius Docker 环境需求

  1. 在主机上安装 Docker.
  2. 为了在 Linux 系统中启用 GPU, 安装 nvidia-docker.

下载 Animius Docker 镜像

官方的 TensorFlow 镜像位于 gundammc/animius Docker Hub 仓库. 镜像的发行版使用以下标签.

每个标签都有一个变体:

例如, 以下命令将下载 Animius 发行版到你的主机:

docker pull gundammc/animius                      # 下载支持 CPU 的最新版本
docker pull gundammc/animius:latest-gpu           # 下载支持 GPU 的最新版本

运行 Animius Docker 容器

使用以下命令来运行 Animius Docker 容器.

docker run [-it] [--rm] [-p hostPort:containerPort] gundammc/animius[:tag] [command]

更多细节请参考: docker run reference.