docker update

2023-05-01 19:13:14 +00:00
parent 27b70ce2be
commit 7f458fded1
2 changed files with 23 additions and 1 deletions
@@ -0,0 +1,21 @@
+name: "cvpr_2023"
+services:
+  lhunor_torch:
+    image: lhunor_torch:v1.13
+    container_name: lhunor_torch_v1.13
+    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]
@@ -10,4 +10,5 @@ pytorch3d
 tables
 numpy-stl
 torchmetrics
-pytorch_msssim
+pytorch_msssim
+importlib