Skip to content

Latest commit

 

History

History
5 lines (4 loc) · 231 Bytes

README.md

File metadata and controls

5 lines (4 loc) · 231 Bytes
Raw

Building Docker Images

Features required for this package:

  • OpenMPI with CUDA support (if CUDA is available for training)
  • PyTorch built with CUDA and OpenMPI (compiled from master) (if CUDA is available for training)