Based on the output it seems your torchvision version was downgraded to 0.2.2:
torchvision {0.8.1 (pytorch/win-64) -> 0.2.2 (pytorch/noarch)}

and I’m unsure why this would happen. Could you create a new virtual environment and reinstall the latest lib versions?



 JamesDickens:

is it worth it to upgrade to 1.7.1.


I would generally recommend to update to the latest release for the best support.
Also, the 1.7.1 binaries ship with cudnn 8.0.5, which should improve the performance compared to 8.0.3.