diff --git a/Dockerfile b/Dockerfile
index 929702a..0bbea19 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -107,6 +107,7 @@ ENV NUM_GPUS=1
 #==2.11.0=cuda*
 #RUN mamba install -y -c conda-forge tensorflow-gpu
 RUN CONDA_OVERRIDE_CUDA="11.2" conda install tensorflow-gpu==2.11.0 -c conda-forge
+RUN CONDA_OVERRIDE_CUDA="11.2" conda install -c conda-forge pytorch-gpu
 WORKDIR /tmp
 RUN mamba install --quiet --yes \
     'notebook' \