BUG dgl wheel on pypi.ngc.nvidia.com is incompatible with PyTorch nightly on aarch64

Hello,

I am running on a new NVIDIA GB10 system and am trying to install DGL in a Python 3.12 virtual environment.

The pre-compiled dgl==2.1.0 wheel available on the pypi.ngc.nvidia.com index is broken and incompatible with the PyTorch nightly builds on the same index.

The Error: After installing torch (version 2.9.0) and dgl (version 2.1.0) from the NVIDIA index, any attempt to import dgl fails with a FileNotFoundError:

FileNotFoundError: Cannot find DGL C++ graphbolt library at /home/bajend01/dgl_gb10_env/lib/python3.12/site-packages/dgl/graphbolt/libgraphbolt_pytorch_2.9.0.so Analysis: The dgl-2.1.0 package is missing the required C++ library (.so file) needed to link against the torch-2.9.0 installation. This indicates the DGL wheel was not correctly built against the PyTorch version it is being distributed with.

The only working solution is to ignore the pre-built wheel and build DGL from source.

Environment Details:

Hardware: NVIDIA GB10

Architecture: aarch64

CUDA Driver: 580.95.05 (from nvidia-smi)

Python: 3.12

PyTorch: torch==2.9.0 (from pypi.ngc.nvidia.com)

DGL: dgl==2.1.0 (from pypi.ngc.nvidia.com)

This appears to be a packaging bug in the aarch64 wheels being provided by NVIDIA.