update docker

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

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

docker/requirements.txt

@@ -4,4 +4,5 @@ scipy
 #deodr
 pandas
 pytorch_msssim
-kornia
+kornia
+wandb