Failed to load image Python extension: Could not find module

vision
20

I had exactly the same specs as you. I solved this issue by creating a new environment in Anaconda and installing the following versions
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch