From 3f60403167435f67ce6d95e388692d5f49795489 Mon Sep 17 00:00:00 2001
From: Hunor Laczko <luni741@gmail.com>
Date: Tue, 9 May 2023 23:04:01 +0000
Subject: [PATCH] update docker

---
 docker/Dockerfile         |  5 +++--
 docker/docker-compose.yml | 20 ++++++++++++++++++++
 docker/requirements.txt   |  3 ++-
 3 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 52044c1..a2cdf54 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,6 +1,7 @@
 # Use an existing image as the base
-#FROM nvcr.io/nvidia/pytorch:22.06-py3
-FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
+# FROM nvcr.io/nvidia/pytorch:23.04-py3
+FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel
+# FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
 
 # Add the current user to the image
 ARG USER_NAME=lhunor
diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 6f00263..240cd90 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -24,6 +24,26 @@ services:
     container_name: lhunor_reconstruction_v1.12
     ipc: host
     tty: true
+    ulimits:
+      memlock: -1
+      stack: 67108864
+    volumes:
+      - '/home-net/lhunor:/home-net'
+      - '/data-net/guest/lhunor/data:/data-net'
+      #- '/data-fast/126-data1/lhunor/data:/data-local/126'
+      - '/data-fast/127-data1/lhunor/data:/data-local/127'
+      - '/data-fast/128-data1/lhunor/data:/data-local/128'
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]
+
+  lhunor_torch_3:
+    image: lhunor_reconstruction:v2.0
+    container_name: lhunor_reconstruction_v2.0
+    ipc: host
+    tty: true
     ulimits:
       memlock: -1
       stack: 67108864
diff --git a/docker/requirements.txt b/docker/requirements.txt
index 334341d..ff217cf 100644
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
@@ -4,4 +4,5 @@ scipy
 #deodr
 pandas
 pytorch_msssim
-kornia
\ No newline at end of file
+kornia
+wandb
\ No newline at end of file