Solved by Ilia_Karmanov in post #7 
              
              
                Thanks for checking Patrick, I’m not sure why it didn’t work for me. 
However I started from scratch and the below works: 
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04

ARG PYTHON_VERSION=3.8
ARG WITH_TORCHVISION=1
ARG pytorch_tag='v1.6.0'
ARG torchvision_tag='v0.7.0'

RUN apt-get update && apt-ge…