Spaces:
Running
Running
File size: 356 Bytes
bc51634 3dd6c96 bc51634 3dd6c96 |
1 2 3 4 5 6 7 8 9 10 11 12 |
# PyTorch Geometric (OCP)
TORCH=2.3.1
CUDA=cu121
pip install --verbose --no-cache torch-scatter -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install --verbose --no-cache torch-sparse -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
# DGL (M3GNet, ALIGNN)
pip install dgl -U -f https://data.dgl.ai/wheels/torch-2.3/cu121/repo.html
|