-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrequirements.txt
33 lines (33 loc) · 957 Bytes
/
requirements.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# pytorch
--find-links https://download.pytorch.org/whl/cu102/torch_stable.html
torch==1.10.2+cu102
--find-links https://download.pytorch.org/whl/cu102/torch_stable.html
torchvision==0.11.3+cu102
--find-links https://download.pytorch.org/whl/cu102/torch_stable.html
torchaudio==0.10.2+cu102
# pytorch_geometric
--find-links https://data.pyg.org/whl/torch-1.10.0+cu102.html
torch-scatter==2.0.9
--find-links https://data.pyg.org/whl/torch-1.10.0+cu102.html
torch-sparse==0.6.12
--find-links https://data.pyg.org/whl/torch-1.10.0+cu102.html
torch-cluster==1.5.9
--find-links https://data.pyg.org/whl/torch-1.10.0+cu102.html
torch-spline-conv==1.2.1
torch-geometric==2.0.3
# other dependencies
ogb==1.3.2
numpy==1.22.2
scipy==1.8.0
scikit-learn==1.0.2
networkx==2.6.3
matplotlib==3.5.1
seaborn==0.11.2
# sulinear_gnn package
-e .
# dev dependencies
jupyterlab==3.2.9
jupyterlab-git==0.34.2
jupyterlab-lsp==3.10.0
jedi-language-server==0.35.1
# CI dependencies