diff --git a/env_setup/Dockerfile b/env_setup/Dockerfile
index a32154ffaff704fe023423bcb5ab99038ee6e1c4..55cb59036db3c3043e866d6c90349cb874004521 100644
--- a/env_setup/Dockerfile
+++ b/env_setup/Dockerfile
@@ -2,6 +2,7 @@
 # Nvidia already has a docker image where they installed all their drivers 
 ###############################################################################
 FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
+ENV CUDA_VISIBLE_DEVICES=0
 
 ###############################################################################
 # Small fix for the internal time of the container. Logging has issues otherwise
@@ -46,4 +47,4 @@ RUN conda init
 # in case you want to run the container as jupyter notebooks.
 
 RUN pip install virtualenv jupyterlab
-CMD export SHELL=/bin/bash && python -m jupyter lab --allow-root --ip='*' --NotebookApp.token='' --NotebookApp.password=''
\ No newline at end of file
+CMD export SHELL=/bin/bash && python -m jupyter lab --allow-root --ip='*' --NotebookApp.token='' --NotebookApp.password=''