Added apptainer container and copying weights of NN

aasensio · Mar 11, 2024 · 9427ca9 · 9427ca9
1 parent 418f2f5
commit 9427ca9
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,6 +4,7 @@ __pycache__/
 *.c
 *.h5
 *.pth
+*.sif
 
 # C extensions
 *.so

diff --git a/MANIFEST.in b/MANIFEST.in
@@ -1,4 +1,5 @@
 include hazel/data/*.1d
+include hazel/data/*.pth
 include hazel/data/LINEAS
 include src/sir/PARAMETER
 include VERSION.txt

diff --git a/apptainer/hazel.def b/apptainer/hazel.def
@@ -0,0 +1,14 @@
+Bootstrap: docker
+From: continuumio/miniconda3
+%post
+    conda update -y conda
+    conda install -y -c conda-forge cython numpy h5py tqdm scipy astropy mpich mpi4py configobj gfortran_linux-64 gcc_linux-64 asciitree matplotlib
+    pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+    pip install torch_geometric
+    pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.0+cpu.html
+    . /opt/conda/etc/profile.d/conda.sh
+    conda activate base 
+    git clone https://github.com/aasensio/hazel2.git
+    cd hazel2
+    python setup.py install 
+    conda clean -afy
diff --git a/apptainer/upload b/apptainer/upload
@@ -0,0 +1 @@
+curl -T hazel.sif --user 'aasensio' 'https://cloud.iac.es/remote.php/dav/files/F00AA55A-3D9E-4B19-A6F7-0EBD138CCF5D/hazel2/hazel.sif' --anyauth