update docker

Dockerfile

@@ -1,15 +0,0 @@
-FROM  pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
-
-RUN apt update && apt upgrade -y
-RUN apt install git -y
-
-RUN git clone https://gitea.lnrweb.com/luni/reconstruction.git
-
-RUN conda install -c fvcore -c iopath -c conda-forge fvcore iopath -y
-RUN conda install -c bottler nvidiacub -y
-RUN conda install pytorch3d -c pytorch3d -y
-
-COPY requirements.txt /tmp/
-RUN pip install --requirement /tmp/requirements.txt
-
-WORKDIR /home/workspace

docker/Dockerfile

@@ -1,5 +1,6 @@
 # Use an existing image as the base
-FROM nvcr.io/nvidia/pytorch:22.06-py3
+#FROM nvcr.io/nvidia/pytorch:22.06-py3
+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

@@ -11,7 +11,27 @@ services:
       - '/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-data1/lhunor/data:/data-local/127'
+      - '/data-fast/128-data1/lhunor/data:/data-local/128'
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]
+
+  lhunor_torch_2:
+    image: lhunor_reconstruction:v1.12
+    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:

docker/requirements.txt

@@ -1,5 +1,7 @@
 plotly
+Cython
 scipy
-deodr
+#deodr
 pandas
-pytorch_msssim
+pytorch_msssim
+kornia

requirements.txt

@@ -1,5 +0,0 @@
-plotly
-scipy
-deodr
-pandas
-pytorch_msssim