🐛[BUG]: DGL build in the Arm container does not work - Githubissues

NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

https://developer.nvidia.com/modulus

Apache License 2.0

798 stars 174 forks source link

🐛[BUG]: DGL build in the Arm container does not work #432

Closed ktangsali closed 2 months ago

ktangsali commented 3 months ago

Version

0.6.0

On which installation method(s) does this occur?

Docker

Describe the issue

For the upcoming release (24.04), we made a switch to the DGL install method here: https://github.com/NVIDIA/modulus/pull/327

While this works for x86 systems, it does not work for the ARM system as DGL currently does not ship wheels for aarch64. Reference: https://[data.dgl.ai/wheels/cu121/repo.html](https://data.dgl.ai/wheels/cu121/repo.html).

Potential fixes include reverting to source build for arm systems.

Minimum reproducible example

No response

Relevant log output

No response

Environment details

No response