Pytorch support SM_87 cuda architecture

1

Can anyone kindly help me download PyTorch, which supports sm_87 on aarch64? Right now, I am using the version below.

import torch
print(torch.version.cuda)
12.2
print(torch.cuda.get_arch_list())
[‘sm_50’, ‘sm_80’, ‘sm_86’, ‘sm_89’, ‘sm_90’, ‘sm_90a’]

2

Check this post.

3

Hi, i tried that and having issue. I have RHEL 9.4

(myenv) [yajuvendra@llmhost pytorch_wheel]$ python -m pip install torch-2.3.0-cp310-cp310-linux_aarch64.whl

ERROR: torch-2.3.0-cp310-cp310-linux_aarch64.whl is not a supported wheel on this platform.

(myenv) [yajuvendra@llmhost pytorch_wheel]$ uname -m

aarch64

(myenv) [yajuvendra@llmhost pytorch_wheel]$ uname -a

Linux llmhost 5.14.0-427.22.1.el9_4.aarch64 #1 SMP PREEMPT_DYNAMIC Mon Jun 10 15:57:50 UTC 2024 aarch64 aarch64 aarch64 GNU/Linux

br…
Yaju

4

Your Python version might be different as the wheel is built for Python 3.10.
If you get stuck, you can use a pre-built container from here.

5

thanks a lot @ptrblck