Pytorch support for sm120

You’re right and I finally got it working. Thanks again for your help.