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
ngc batch run -n "Cloud-nv-us-west-2-633840" -i "nvidia/pytorch:18.01-py3" --ace nv-us-west-2 -in ngcv8 -c "python /mnt/code/main.py --train_img --train_CNN --train_img_dir /mnt/VQA_1/train2014/ --val_img_dir /mnt/VQA_1/val2014/ --data_dir /mnt/data/ --show_batch_res --batch_size 600 --cnn_type resnext-101-modular --cuda --pretrained_model_file /mnt/pretrain/resnext_101_32x4d.pth" --result /results --datasetid 8556:/mnt/code --datasetid 8478:/mnt/data --datasetid 8322:/mnt/VQA_1 --datasetid 8558:/mnt/pretrain