pytorch-cuda Docker image based on nvidia/cuda images along pytorch and python. Current master version cuda==11.0 cudnn==8 python==3.6.5 pytorch==1.7.0 torchaudio==0.7.0