init

.woodpecker.yml

@@ -7,7 +7,7 @@ steps:
     settings:
       registry: git.sandbox.iuk.hdm-stuttgart.de
       repository: grosse/jupyterlab-datascience-gpu
-      tag: 4.0.x-gpu-torch
+      tag: 4.0.x-gpu-ollama
       architectures: amd64
       context: Dockerfile
       loglevel: debug

Dockerfile

@@ -1,15 +1,11 @@
 FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12d
-# debian:bullseye-20230109-slim
+
 ARG NB_USER="jovyan"
 ARG NB_UID="1000"
 ARG NB_GID="100"
 ENV CONDA_CUDA_OVERRIDE="12.1"  
 ENV NUM_GPUS=1
 USER root
-RUN apt-get update --yes && \
-    apt-get install --yes --no-install-recommends \
-    ffmpeg  \
-    libportaudio2  && \
-    apt-get clean && rm -rf /var/lib/apt/lists/* 
+RUN curl -fsSL https://ollama.com/install.sh | sh
+RUN command -v systemctl >/dev/null && sudo systemctl stop ollama
 USER $NB_UID
-RUN CONDA_OVERRIDE_CUDA="12.1" conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia