diff --git a/slurm/build_apptainer.sh b/slurm/build_apptainer.sh index 3bd2518..1f40b69 100755 --- a/slurm/build_apptainer.sh +++ b/slurm/build_apptainer.sh @@ -6,7 +6,7 @@ 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 \ . @@ -14,5 +14,8 @@ docker build \ 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 index 4bd8f22..f5a4e44 100644 --- 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