/pytorch-build-mpi-cuda

Docker image for pytorch with openmpi and cuda support

Primary LanguageDockerfile

pytorch-build-mpi-cuda

Features:

  • openmpi with cuda support
  • pytorch with cuda and openmpi (compiled from master)
  • apex