This is a copy of nvcr.io/nvidia/pytorch:18.05-py3 - the only difference is we've run python /workspace/examples/mnist/main.py once to pre-download the example dataset to avoid download waits and networking hazards during the demo. For more info on the original demo, see https://developer.nvidia.com/blog/gpu-containers-runtime/.
Content type
Image
Digest
Size
1.9 GB
Last updated
over 5 years ago
docker pull mirantistraining/gpu-demo:pytorch-18.05-py3