iterait/tensorflow Docker 镜像 - 轩辕镜像 | Docker 镜像高效稳定拉取服务

Docker镜像

这是Iterait a.s.提供的开源Docker镜像仓库。

我们使用nvidia-docker而非常规docker以增强GPU支持，但也可使用常规docker。

DockerHub

所有构建好的镜像均存储在Iterait DockerHub。

构建

所有镜像均可通过常规docker构建流程构建。

bash
$ docker build -t <image-name> -f <dockerfile> --squash .

如果需要NVIDIA CUDA支持，请使用tag=cuda构建参数构建镜像。

bash
$ docker build --build-arg tag=cuda -t <image-name>:cuda -f <dockerfile> .

运行

无GPU支持的容器运行非常简单。

bash
$ docker run -it <image-name> /bin/bash

要运行带GPU支持的容器，建议使用nvidia-docker。以下示例展示了使用两块GPU并执行bash的方法。

bash
$ docker run --runtime=nvidia \
    -e NVIDIA_VISIBLE_DEVICES=0,1 \
    -it <image-name>:cuda /bin/bash