Bootstrap: docker
From: nvidia/cuda:10.1-devel-centos7

%post
    yum -y install python3
    pip3 install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html