From f5fec02bec258a1327c5ca2387fba10d5bf2fb6b Mon Sep 17 00:00:00 2001
From: maltegrosse <malte.grosse@googlemail.com>
Date: Tue, 9 May 2023 17:03:32 +0900
Subject: [PATCH] added pytorch

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

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' \