all in

Author: janpf

.github/workflows/dockerimage.yml

@@ -5,7 +5,7 @@ name: Create and publish a Docker image
 on:
   push:
     branches:
-      - main
+      - germeval25
 
 # Defines two custom environment variables for the workflow. These are used for the Container registry domain, and a name for the Docker image that this workflow builds.
 env:
@@ -50,7 +50,7 @@ jobs:
         uses: docker/metadata-action@9ec57ed1fcdbf14dcef7dfbe97b2010124a938b7
         with:
           images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
-          tags: main
+          tags: ${{ github.ref_name }}
       # This step uses the `docker/build-push-action` action to build the image, based on your repository's `Dockerfile`. If the build succeeds, it pushes the image to GitHub Packages.
       # It uses the `context` parameter to define the build's context as the set of files located in the specified path. For more information, see "[Usage](https://github.com/docker/build-push-action#usage)" in the README of the `docker/build-push-action` repository.
       # It uses the `tags` and `labels` parameters to tag and label the image with the output from the "meta" step.

k8s/templates/Dockerfile

@@ -1,5 +1,4 @@
-FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
-#FROM nvidia/cuda:12.3.1-devel-ubuntu22.04
+FROM nvidia/cuda:12.8.1-devel-ubuntu22.04
 
 ARG USER={{lastname}}
 ARG UID={{uuid}}
@@ -33,8 +32,8 @@ RUN chown -R ${USER}:${USER} /localdir/
 
 COPY . .
 RUN python -m pip install packaging wheel
-RUN python -m pip install --extra-index-url https://download.pytorch.org/whl/cu121 torch==2.5.1+cu121
-RUN python -m pip install --no-build-isolation --extra-index-url https://download.pytorch.org/whl/cu121 -r requirements.txt
+RUN python -m pip install --extra-index-url https://download.pytorch.org/whl/cu128 torch==2.7.1+cu128
+RUN python -m pip install --no-build-isolation --extra-index-url https://download.pytorch.org/whl/cu128 -r requirements_core.txt
 Run python -m pip freeze
 
 #RUN python -m pip install flash-attn==2.5.7 --no-build-isolation --extra-index-url https://download.pytorch.org/whl/cu121 -r requirements.txt

requirements_core.txt

@@ -13,6 +13,6 @@ safetensors
 sentence-transformers==3.3.1
 sentencepiece
 tokenizers
-torch==2.5.1+cu121
+torch==2.7.1+cu128
 transformers
 uv