diff --git a/Dockerfile b/Dockerfile
index 37e8c6e..b1d6f8c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -17,6 +17,8 @@ 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
 USER jovyan
 
-# https://anaconda.org/pytorch-test/pytorch-cuda
\ No newline at end of file
+# https://anaconda.org/pytorch-test/pytorch-cuda
+#https://github.com/NVIDIA/nvidia-docker/issues/871#issuecomment-966668913
\ No newline at end of file