GNNRecom/requirements_cuda.txt

16 lines
301 B
Plaintext

-f https://data.dgl.ai/wheels/repo.html
dgl==0.7.0-cu110
django>=3.2.8
gensim>=3.8.3
matplotlib>=3.3.1
mysqlclient>=2.0.3
numpy>=1.20.1
ogb>=1.2.5
pandas>=1.2.2
scikit-learn>=0.24.1
scipy>=1.6.1
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.7.1+cu110
tqdm>=4.57.0
transformers>=4.2.2