Hi AgitatedDove14 , i changed everything to cuda 10.1 and tried again with the same rrror. the section as follows. I made sure torch==1.6.0+cu101 and torchvision==0.8.2+cu101 are in the pypi repo. But the same error still came up.
` # Python 3.6.9 (default, Oct 8 2020, 12:12:24) [GCC 8.4.0]
boto3 == 1.14.56
clearml == 0.17.4
numpy == 1.19.1
torch == 1.6.0
torchvision == 0.7.0

Detailed import analysis

**************************

IMPORT PACKAGE boto3

clearml.storage: 0

IMPORT PACKAGE clearml

pytorch_mnist.py: 14

IMPORT PACKAGE numpy

pytorch_mnist.py: 13

IMPORT PACKAGE torch

pytorch_mnist.py: 8,9,10,11

IMPORT PACKAGE torchvision

pytorch_mnist.py: 12 `