2024-03-18 02:37:17 +00:00
|
|
|
FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12d
|
2023-02-01 09:05:43 +00:00
|
|
|
# debian:bullseye-20230109-slim
|
|
|
|
|
ARG NB_USER="jovyan"
|
|
|
|
|
ARG NB_UID="1000"
|
2024-03-18 01:30:04 +00:00
|
|
|
ARG NB_GID="100"
|
2024-03-18 13:00:47 +00:00
|
|
|
ENV CONDA_CUDA_OVERRIDE="11.8"
|
2024-03-18 01:30:04 +00:00
|
|
|
ENV NUM_GPUS=1
|
2024-03-18 06:02:01 +00:00
|
|
|
USER root
|
2024-03-18 06:12:21 +00:00
|
|
|
RUN apt-get update --yes && \
|
|
|
|
|
apt-get install --yes --no-install-recommends \
|
2024-03-18 06:02:01 +00:00
|
|
|
ffmpeg \
|
2024-03-18 06:23:07 +00:00
|
|
|
libportaudio2 && \
|
2024-03-18 06:02:01 +00:00
|
|
|
apt-get clean && rm -rf /var/lib/apt/lists/*
|
|
|
|
|
USER $NB_UID
|
2024-03-18 13:00:47 +00:00
|
|
|
#RUN pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
|
|
|
|
|
#RUN CONDA_OVERRIDE_CUDA="12.1" conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
|
|
|
|
|
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
|
