This defines PyTorch Docker image used in Walker Lab.
Docker image for scientific computing with PyTorch