Niksa Praljak
commited on
Commit
•
9599488
1
Parent(s):
b497299
Add installation scripts/files for downloading
Browse files- requirements.txt +38 -0
- torch_requirements.sh +3 -0
requirements.txt
ADDED
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Foundational Libraries (core dependencies)
|
2 |
+
setuptools==59.5.0
|
3 |
+
numpy==1.23.5 # Pin numpy to ensure compatibility
|
4 |
+
scipy==1.9.3 # Install scipy after numpy
|
5 |
+
pandas==1.5.3 # Pin pandas to avoid dask incompatibilities
|
6 |
+
|
7 |
+
# Libraries dependent on numpy, pandas, or scipy
|
8 |
+
numba==0.58.1
|
9 |
+
dask==2023.4.1 # Use a stable version compatible with pandas==1.5.3
|
10 |
+
scikit-learn==1.2.2
|
11 |
+
imageio==2.34.0
|
12 |
+
matplotlib==3.7.1
|
13 |
+
seaborn==0.12.2
|
14 |
+
wordcloud==1.9.3
|
15 |
+
|
16 |
+
# Machine Learning and Deep Learning
|
17 |
+
torchmetrics==0.10.3
|
18 |
+
pytorch-lightning==1.9.5
|
19 |
+
transformers==4.29.2
|
20 |
+
deepspeed==0.9.2
|
21 |
+
accelerate==0.19.0
|
22 |
+
diffusers==0.16.1
|
23 |
+
optuna==3.1.1
|
24 |
+
fair-esm==2.0.0
|
25 |
+
|
26 |
+
# Specialized Libraries
|
27 |
+
lightning-flash==0.8.2
|
28 |
+
axial-positional-embedding==0.2.1
|
29 |
+
linear-attention-transformer==0.19.1
|
30 |
+
linformer==0.2.3
|
31 |
+
|
32 |
+
# Bioinformatics
|
33 |
+
biopython==1.79
|
34 |
+
|
35 |
+
# General Utilities
|
36 |
+
tqdm==4.65.0
|
37 |
+
pynvml==11.5.0
|
38 |
+
|
torch_requirements.sh
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
#!/bin/bash
|
2 |
+
|
3 |
+
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
|