Trying to install pytorch with GPU

6

As henry said using pip to install pytorch worked.I had to do some digging to find the relevant version.Seems cuda 9 uses pytorch 1.1 and torchvision 0.3

1 Like