Skip to content

Commit

Permalink
Singularity definitions
Browse files Browse the repository at this point in the history
  • Loading branch information
truatpasteurdotfr committed Oct 5, 2022
1 parent ca761da commit e01f546
Show file tree
Hide file tree
Showing 2 changed files with 46 additions and 0 deletions.
16 changes: 16 additions & 0 deletions Singularity.from.ghcr.io
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
Bootstrap: docker
From: ghcr.io/truatpasteurdotfr/model-angelo:main

%runscript
# do not use ~/.local python
PYTHONNOUSERSITE=1
export PYTHONNOUSERSITE

export TORCH_HOME=/public/model_angelo_weights

eval "$(conda shell.bash hook)"
conda activate model_angelo

#source `which activate` model_angelo
model_angelo "$@"

30 changes: 30 additions & 0 deletions Singularity.from.scratch
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
Bootstrap: docker
From: continuumio/miniconda3

%post

conda update conda
conda upgrade --all -y

# https://github.com/3dem/model-angelo
cd /opt && \
git clone https://github.com/3dem/model-angelo.git && \
export TORCH_HOME=/public/model_angelo_weights && \
mkdir -p /public/model_angelo_weights && \
cd model-angelo && \
bash install_script.sh --download-weights


%runscript
# do not use ~/.local python
PYTHONNOUSERSITE=1
export PYTHONNOUSERSITE

export TORCH_HOME=/public/model_angelo_weights

eval "$(conda shell.bash hook)"
conda activate model_angelo

#source `which activate` model_angelo
model_angelo "$@"

0 comments on commit e01f546

Please sign in to comment.