Skip to content

Latest commit

 

History

History
73 lines (60 loc) · 3.3 KB

README.md

File metadata and controls

73 lines (60 loc) · 3.3 KB
Raw

Docker for users

We provide Dockerfiles in order to build containerized execution environment that ease the use of Ignite for computer vision and NLP tasks.

These images are also provided with the following Horovod configuration:

Available Frameworks:
    [ ] TensorFlow
    [X] PyTorch
    [ ] MXNet

Available Controllers:
    [ ] MPI
    [X] Gloo

Available Tensor Operations:
    [X] NCCL
    [ ] DDL
    [ ] CCL
    [ ] MPI
    [X] Gloo

Installation

  • main/Dockerfile.base: latest stable PyTorch, Ignite with minimal dependencies
    • docker pull pytorchignite/base:latest
  • main/Dockerfile.vision: base image with useful computer vision libraries
    • docker pull pytorchignite/vision:latest
  • main/Dockerfile.nlp: base image with useful NLP libraries
    • docker pull pytorchignite/nlp:latest
  • main/Dockerfile.apex: multi-stage NVIDIA/apex build with latest Pytorch, Ignite image with minimal dependencies
    • docker pull pytorchignite/apex:latest
  • main/Dockerfile.apex-vision: base apex with useful computer vision libraries
    • docker pull pytorchignite/apex-vision:latest
  • main/Dockerfile.apex-nlp: base apex with useful NLP libraries
    • docker pull pytorchignite/apex-nlp:latest
  • hvd/Dockerfile.hvd-base: multi-stage Horovod build with latest stable PyTorch, Ignite with minimal dependencies
    • docker pull pytorchignite/hvd-base:latest
  • hvd/Dockerfile.hvd-vision: base Horovod image with useful computer vision libraries
    • docker pull pytorchignite/hvd-vision:latest
  • hvd/Dockerfile.hvd-nlp: base Horovod image with useful NLP libraries
    • docker pull pytorchignite/hvd-nlp:latest
  • hvd/Dockerfile.hvd-apex: multi-stage NVIDIA/apex and Horovod build with latest Pytorch, Ignite image with minimal dependencies
    • docker pull pytorchignite/hvd-apex:latest
  • hvd/Dockerfile.hvd-apex-vision: base Horovod apex with useful computer vision libraries
    • docker pull pytorchignite/hvd-apex-vision:latest
  • hvd/Dockerfile.hvd-apex-nlp: base Horovod apex with useful NLP libraries
    • docker pull pytorchignite/hvd-apex-nlp:latest

Deprecated images (no version updates)

How to use

docker run -it -v $PWD:/workspace/project --network=host --ipc=host pytorchignite/base:latest /bin/bash

Building the image yourself

Dockerfiles are supplied to build images with dependencies that ease the use of Ignite for computer vision / NLP tasks:

cd main
docker build -t pytorchignite/base:latest -f Dockerfile.base .