grosse
/
jupyterlab-datascience-gpu
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

init
ci/woodpecker/push/woodpecker Pipeline failed Details

This commit is contained in:
Malte Grosse 2024-03-18 19:59:38 +09:00
parent dc259d2af9
commit 519c934876
2 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.woodpecker.yml
View File

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

10
Dockerfile
View File

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