Solved by ptrblck in post #2 
              
              
                Since you are using an Ampere GPU (3070), you would need to use CUDA>=11.0, so the old PyTorch 1.5.1 release with CUDA9.2 won’t work. 
Update to the latest release with CUDA11.1 and it should work.