Yeah, you are right. Apparently I was looking at the 710m.
Since Pytorch needs compute capability >=3.0, you should be fine.
What does torch.cuda.is_available() return?
Have you tried to reinstall Pytorch?