This no longer works (or perhaps the object is stripped), @tom shared the latest approach on slack and it’s:
$ python -c "import torch; print(torch.cuda.get_arch_list())"
['sm_37', 'sm_50', 'sm_60', 'sm_70', 'sm_75', 'sm_80']
Thank you, @tom.
2 Likes