jupyterlab-datascience-gpu/Dockerfile

8 lines
294 B
Docker

FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12f
# debian:bullseye-20230109-slim
ARG NB_USER="jovyan"
ARG NB_UID="1000"
ARG NB_GID="100"
ENV CONDA_CUDA_OVERRIDE="12.0"
ENV NUM_GPUS=1
RUN CONDA_OVERRIDE_CUDA="12.0" conda install tensorflow-gpu==2.15.0 -c conda-forge