Solved by ptrblck in post #2 
              
              
                The PyTorch binaries ship with CUDA for GPUs with compute capability >=3.7, while your GPU has 3.5, so you would need to build from source as shown here. 
Let us know, if you get stuck somewhere.