Thanks, compiling from source worked and in fact it detected the 2.1 variant. I didn’t know it exists.
Now i get this error when using torch.nn.UpsamplingBilinear2d:
RuntimeError: cuda runtime error (7) : too many resources requested for launch at /home/user/pytorch/torch/lib/THCUNN/generic/SpatialUpSamplingBilinear.cu:63
Is it still related to the oldness of my GPU?