diff --git a/Dockerfile b/Dockerfile
index 33112e1..e0f2e2c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,7 +3,7 @@ FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12d
 ARG NB_USER="jovyan"
 ARG NB_UID="1000"
 ARG NB_GID="100"
-ENV CONDA_CUDA_OVERRIDE="12.1"  
+ENV CONDA_CUDA_OVERRIDE="11.8"  
 ENV NUM_GPUS=1
 USER root
 RUN apt-get update --yes && \
@@ -12,4 +12,6 @@ RUN apt-get update --yes && \
     libportaudio2  && \
     apt-get clean && rm -rf /var/lib/apt/lists/* 
 USER $NB_UID
-RUN CONDA_OVERRIDE_CUDA="12.1" conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
\ No newline at end of file
+#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
\ No newline at end of file