Refine sif building

pwr-ai · Dec 13, 2024 · 72ac731 · 72ac731
1 parent 795ebf0
commit 72ac731
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/slurm/build_apptainer.sh b/slurm/build_apptainer.sh
@@ -6,13 +6,16 @@ set -e
 
 # Creates docker image containing the environment for fine-tuning.
 docker build \
-    --tag juddges_sft:latest \
+    --tag juddges:latest \
     --file ./slurm/fine_tuning_env.dockerfile \
     .
 
 # Converts the docker image into an Apptainer image.
 docker run \
     --rm \
     --volume /var/run/docker.sock:/var/run/docker.sock \
-    --volume $(pwd):/juddges \
-    kaczmarj/apptainer build juddges_sft.sif docker-daemon://juddges_sft:latest
+    --volume $(pwd):/work \
+    kaczmarj/apptainer \
+        build \
+        juddges.sif \
+        docker-daemon://juddges:latest
diff --git a/slurm/fine_tuning_env.dockerfile b/slurm/fine_tuning_env.dockerfile
@@ -1,6 +1,6 @@
 FROM nvcr.io/nvidia/pytorch:24.11-py3
 
-WORKDIR /judddges
+WORKDIR /juddges
 
 RUN apt-get update -qq && apt-get install --yes -q make git