small updates

docker/Dockerfile2

@@ -42,6 +42,7 @@ COPY requirements.txt /tmp/
 RUN pip install torch-cluster -f https://data.pyg.org/whl/torch-2.0.0+cu117.html
 RUN pip install --requirement /tmp/requirements.txt
 
+RUN pip install pyvista flask==2.1.2 pyinstaller==5.1 sphinx ipyvtklink pythreejs tqdm imageio imageio-ffmpeg matplotlib cmocean
 
 # defining the volumes that will be mounted
 VOLUME [ "/data-net" ]

docker/docker-compose.yml

@@ -48,6 +48,48 @@ services:
     container_name: lhunor_iccvw_pt2.0.1
     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/127-data2/lhunor/data:/data-local/127_2'
+      - '/data-fast/128-data1/lhunor/data:/data-local/128'
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]
+
+  lhunor_iccvw_2:
+    image: lhunor_iccvw:pt2.0.1
+    container_name: lhunor_iccvw_2_pt2.0.1
+    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/127-data2/lhunor/data:/data-local/127_2'
+      - '/data-fast/128-data1/lhunor/data:/data-local/128'
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]
+
+  lhunor_iccvw_pv:
+    image: lhunor_iccvw:pt2.0.1_pv
+    container_name: lhunor_iccvw_pt2.0.1_pv
+    ipc: host
+    tty: true
     ulimits:
       memlock: -1
       stack: 67108864