docker pull nvcr.io/nvidia/pytorch:18.01-py3
nvidia-docker run -it --name sai nvcr.io/nvidia/pytorch:18.01-py3 /bin/bash
(Just try to import torch in python shell here). Exit with Ctrl+D.
docker commit -m test -a sai pytorch_sai nvcr.io/mila1234/pytorch_sai:1
docker push nvcr.io/mila1234/pytorch_sai:1