Hmm, it doesn’t seem to install the cuda version when I do a pip install torchvision
I event tried to do a pip install torchvision==0.15.1+cu118
Uninstalled it:
Successfully uninstalled torchvision-0.15.1
PS C:\Users\Dusti> pip install torchvision==0.15.1+cu118
ERROR: Could not find a version that satisfies the requirement torchvision==0.15.1+cu118 (from versions: 0.1.6, 0.1.7, 0.1.8, 0.1.9, 0.2.0, 0.2.1, 0.2.2, 0.2.2.post2, 0.2.2.post3, 0.12.0, 0.13.0, 0.13.1, 0.14.0, 0.14.1, 0.15.0, 0.15.1)
ERROR: No matching distribution found for torchvision==0.15.1+cu118