diff --git a/Dockerfile b/Dockerfile
index e376062..d928808 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,4 +2,7 @@ FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12c
 # debian:bullseye-20230109-slim
 ARG NB_USER="jovyan"
 ARG NB_UID="1000"
-ARG NB_GID="100"
\ No newline at end of file
+ARG NB_GID="100"
+ENV CONDA_CUDA_OVERRIDE="11.2"  
+ENV NUM_GPUS=1
+RUN CONDA_OVERRIDE_CUDA="11.2" conda install -c conda-forge pytorch-gpu
\ No newline at end of file