The current PyTorch pip wheels and conda binaries ship with a min.compute capability of 3.7, so you would need to build PyTorch from source for your GPU as described here 51. You might be able to use an older release (I believe ~1.5 might have used 3.5 as the min. compute capability, but I’m not sure), but you would miss the latest features and fixes.