Solved by ptrblck in post #4 
              
              
                The latest stable PyTorch version is 1.6.0 and you can find the install commands for the conda binaries or pip wheels here. I would recommend to use the latest CUDA version (10.2), but make sure your driver supports it (you are currently using 10.1 so you might use an older driver). 
You can see the…