From 519c9348769a4b415920cd4723517a56ee01dde4 Mon Sep 17 00:00:00 2001
From: Malte Grosse <malte.grosse@googlemail.com>
Date: Mon, 18 Mar 2024 19:59:38 +0900
Subject: [PATCH] init

---
 .woodpecker.yml |  2 +-
 Dockerfile      | 10 +++-------
 2 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/.woodpecker.yml b/.woodpecker.yml
index 2fcb81c..ea2b7be 100644
--- a/.woodpecker.yml
+++ b/.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
diff --git a/Dockerfile b/Dockerfile
index 33112e1..55634b9 100644
--- a/Dockerfile
+++ b/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
\ No newline at end of file