Can’t found PyTorch2.0 with Cuda on aarch64

Where can I find pytorch2-py39-cuda-aarch64?
(Jetson AGX Orin)
Thank you very much.

1 Like

Hi @yeh0102, we provide pre-built PyTorch wheels for the default version of Python that comes with Ubuntu, and on JetPack 5 / Ubuntu 20.04 that is Python 3.8 (on JetPack 4 / Ubuntu 18.04 that’s Python 3.6). So for Python 3.9 you would need to build it from source. There are build instructions in this thread that you can loosely follow: PyTorch for Jetson (you may still need to apply this patch)

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.