diff --git a/Dockerfile b/Dockerfile
index b1d6f8c..d833780 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,6 +18,7 @@ RUN apt update && apt install -y libcudnn8
 #RUN pip install tensorflow
 RUN pip install --force-reinstall tf-nightly==2.12.0.dev20230126
 RUN pip install --upgrade tensorrt
+RUN pip install pytorch
 USER jovyan
 
 # https://anaconda.org/pytorch-test/pytorch-cuda