requirements.txt 265 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
absl-py==0.13.0
biopython==1.79
deepspeed==0.5.3
dm-tree==0.1.6
ml-collections==0.1.0
numpy==1.21.2
PyYAML==5.4.1
requests==2.26.0
scipy==1.7.1
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
10
torch==1.10.0
11
12
tqdm==4.62.2
typing-extensions==3.10.0.2
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
13
pytorch_lightning==1.5.0
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
14
git+git://github.com/NVIDIA/dllogger.git