From 637c0fb0f71a551042762c13f76c68bf8953dda0 Mon Sep 17 00:00:00 2001
From: maltegrosse <malte.grosse@googlemail.com>
Date: Fri, 27 Jan 2023 12:56:36 +0100
Subject: [PATCH] added pytorch

---
 Dockerfile | 1 +
 1 file changed, 1 insertion(+)

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