found diff

2023-03-30 13:44:59 +00:00
parent 0a65dbd75d
commit a27f5caf0c
4 changed files with 38 additions and 10 deletions
@@ -0,0 +1,21 @@
+services:
+  lhunor_torch:
+    image: lhunor_smpl:v2
+    container_name: lhunor_smpl
+    #command: python -c "import tensorflow as tf;tf.test.gpu_device_name()"
+    ipc: host
+    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'
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]