Pytorch 2.0.0 & Cuda 11.4 on aarch64

Question

Pytorch 2.0.0 & Cuda 11.4 on aarch64

YoushaaMurhij opened this issue 10 months ago · comments

Hi,
I am trying to build torch-scatter on aarch64 with cuda11.4 and pytorch 2.0 using this command:

RUN pip3 install torch-scatter==2.1.1 -f https://data.pyg.org/whl/torch-2.0.0+cu114.html

But I got this error message while running the application:

RuntimeError: Not compiled with CUDA support

Could you please guide me how to build it correctly?
Thanks

Matthias Fey · Answer 1 · Tue Jul 25 2023 22:02:17 GMT+0800 (China Standard Time)

There does not exist a https://data.pyg.org/whl/torch-2.0.0+cu114.html version. Try https://data.pyg.org/whl/torch-2.0.0+cu117.html

Youshaa Murhij · Answer 2 · Tue Jul 25 2023 22:06:17 GMT+0800 (China Standard Time)

I am trying to build pytorch scatter on jetson xavier. It has cuda 11.4. I do not think it is an option for me to upgrade cuda.
does building from main branch with force cuda flag work? I think, I can change pytorch version.

Matthias Fey · Answer 3 · Tue Jul 25 2023 22:39:09 GMT+0800 (China Standard Time)

Yes, you can also try FORCE_CUDA=1 pip install torch-scatter.

Youshaa Murhij · Answer 4 · Wed Jul 26 2023 00:30:39 GMT+0800 (China Standard Time)

Yes, it works for me on aarch with cuda11.4.
Thanks for your response!