Niksa Praljak commited on
Commit
9599488
1 Parent(s): b497299

Add installation scripts/files for downloading

Browse files
Files changed (2) hide show
  1. requirements.txt +38 -0
  2. 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