From 1c2ed4456379c2a30e5e6740df8cf883b89762a9 Mon Sep 17 00:00:00 2001
From: Malte Grosse <malte.grosse@googlemail.com>
Date: Mon, 18 Mar 2024 10:30:04 +0900
Subject: [PATCH] add torch

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

diff --git a/Dockerfile b/Dockerfile
index e376062..d928808 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,4 +2,7 @@ FROM git.sandbox.iuk.hdm-stuttgart.de/grosse/jupyterlab-datascience:4.0.12c
 # debian:bullseye-20230109-slim
 ARG NB_USER="jovyan"
 ARG NB_UID="1000"
-ARG NB_GID="100"
\ No newline at end of file
+ARG NB_GID="100"
+ENV CONDA_CUDA_OVERRIDE="11.2"  
+ENV NUM_GPUS=1
+RUN CONDA_OVERRIDE_CUDA="11.2" conda install -c conda-forge pytorch-gpu
\ No newline at end of file