-
Notifications
You must be signed in to change notification settings - Fork 6
/
requirements.txt
50 lines (42 loc) · 1.15 KB
/
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# PyTorch.
torch==1.10.1+cu113
torchvision==0.11.2+cu113
torchaudio==0.10.1+cu113
# PyTorch Geometric.
torch-scatter==2.0.9
torch-sparse==0.6.13
torch-cluster==1.6.0
torch-spline-conv==1.2.1
torch-geometric==2.0.4
# Hugging Face.
transformers==4.16.2
tokenizers==0.11.4
datasets==1.18.3
# Optimization.
deepspeed==0.5.10 #NB: Change line 1717 of 'deepspeed/runtime/zero/stage_1_and_2.py' from "bit16_partitions[partition_id]" to "[bit16_partitions[partition_id]]".
git+https://github.com/luyug/GradCache@main#egg=GradCache
# Visualisation.
matplotlib==3.2.2
seaborn==0.11.2
plotly==5.8.0
kaleido==0.2.1
pyvis==0.2.0
# Other (ML).
numpy==1.21.0
scikit-learn==1.0.2
pytorch-lightning==1.5.10
sentence-transformers==2.1.0
wandb==0.12.10
networkx==2.6.3
node2vec==0.4.3
sentencepiece==0.1.97
easynmt==2.0.2
# Other (data).
pandas==1.3.5
spacy==3.3.0
fr_core_news_md @ https://github.com/explosion/spacy-models/releases/download/fr_core_news_md-3.3.0/fr_core_news_md-3.3.0-py3-none-any.whl
# Other.
jupyter==1.0.0
# Package locations.
--find-links https://download.pytorch.org/whl/cu113/torch_stable.html
--find-links https://data.pyg.org/whl/torch-1.10.0+cu113.html