ptrblck:

PyTorch 1.5.0 and torchvision 0.6.0 should be compatible.
Are you using the 1.0.0 version by any chance?


My torch version in python is 1.0.0 but in conda it is 1.5.0. How is it possible ? I guess the problem comes from my torch version if 1.0.0 is actually my torch version.