new docker compose

cvg · Nov 2, 2023 · eb47cfe · eb47cfe
1 parent 9b461eb
commit eb47cfe
Showing 1 changed file with 33 additions and 6 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -13,19 +13,46 @@ services:
     volumes:
     - '/tmp/.X11-unix:/tmp/.X11-unix'
     - ${SSH_AUTH_SOCK}:/ssh-agent
+    - /dev/shm:/dev/shm
+    # - '/:/host:ro'
+    # - './inference.py:/OnePose_Plus_Plus_Spot/inference.py'
+    - './outputs:/OnePose_Plus_Plus_Spot/outputs'
+    devices:
+    - /dev/dri
+    privileged: true
+    runtime: nvidia
+    shm_size: '50gb'
+    image: registry.ethz.ch/cvg/robotics/spot_pose_estimation:hermann
+    command: python3 /OnePose_Plus_Plus_Spot/run.py
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities:
+                - gpu
+              driver: nvidia
+  spot_pose_estimation_train:
+    network_mode: host
+    environment:
+    - DISPLAY=$DISPLAY
+    - QT_X11_NO_MITSHM=1
+    - XAUTHORITY=/tmp/.docker.xauth
+    - NVIDIA_VISIBLE_DEVICES=all
+    - NVIDIA_DRIVER_CAPABILITIES=all
+    volumes:
+    - '/tmp/.X11-unix:/tmp/.X11-unix'
     - '/:/host:ro'
     - './inference.py:/OnePose_Plus_Plus_Spot/inference.py'
     - './outputs:/OnePose_Plus_Plus_Spot/outputs'
-    # - './SpotRobot-annotate:/scratch_net/biwidl215/myamaguchi/OnePose_Plus_Plus/data/demo/onepose_6/onepose_6-annotate'
-    # - './ros2/spot_pose_estimation:/ros2_ws/src/spot_pose_estimation'
-    # - '/media/blumh/data/spot_pose_estimation/spot-synthetic-dataset:/data'
-    # - './sfm_model:/OnePose_Plus_Plus_Spot/data'
+    - './ros2/spot_pose_estimation:/ros2_ws/src/spot_pose_estimation'
+    - './data:/data'
+    - './sfm_model:/OnePose_Plus_Plus_Spot/data'
     devices:
     - /dev/dri
     privileged: true
     runtime: nvidia
-    shm_size: '50gb'
-    image: spot_pose_estimation
+    shm_size: '70gb'
+    image: registry.ethz.ch/cvg/robotics/spot_pose_estimation:hermann
     command: python3 /OnePose_Plus_Plus_Spot/run.py
     deploy:
       resources: