mlip-arena / scripts /install-dgl.sh
cyrusyc's picture
avoid update torch during dgl installation
aeaedb4
raw
history blame
124 Bytes
# DGL (M3GNet, ALIGNN)
TORCH=2.2
CUDA=cu121
pip install dgl -f https://data.dgl.ai/wheels/torch-${TORCH}/${CUDA}/repo.html