Stonesjtu/pytorch-build-mpi-cuda

Docker image for pytorch with openmpi and cuda support

Dockerfile