add docker-compose

2023-05-23 10:35:54 +00:00
parent e108460c48
commit b5b2f1616e
1 changed files with 24 additions and 0 deletions
@@ -0,0 +1,24 @@
+name: "lhunor_iccv23"
+services:
+  lhunor_tf:
+    image: lhunor_tf:2023.02.21
+    container_name: lhunor_iccv23
+    #command: python -c "import tensorflow as tf;tf.test.gpu_device_name()"
+    ipc: host
+    user: 11017:11000
+    tty: true
+    ulimits:
+      memlock: -1
+      stack: 67108864
+    volumes:
+      - '~/workspace:/workspace'
+      - '/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'
+    working_dir: /data-net
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]