Building Docker Images
January 11, 2021 ยท View on GitHub
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)