From abb6c34bd6b78642c56b7914d97b9bfee5ed107a Mon Sep 17 00:00:00 2001
From: maltegrosse <malte.grosse@googlemail.com>
Date: Fri, 27 Jan 2023 12:11:33 +0100
Subject: [PATCH] added tensor rt

---
 Dockerfile | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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