jupyterlab-datascience-gpu/Dockerfile

8 lines
294 B
Docker
Raw Permalink Normal View History

2024-06-01 09:57:18 +00:00
FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12f
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 04:56:40 +00:00
ENV CONDA_CUDA_OVERRIDE="12.0"
2024-03-18 01:30:04 +00:00
ENV NUM_GPUS=1
2024-03-18 04:56:40 +00:00
RUN CONDA_OVERRIDE_CUDA="12.0" conda install tensorflow-gpu==2.15.0 -c conda-forge