Fix simulation docker-compose file with slurm (#141)

* fix simulation docker-compose file with slurm * Force workflow rerun * Update setup.py --------- Co-authored-by: VishGit1234 <[email protected]>
WATonomous · Jul 29, 2024 · 22f02c6 · 22f02c6
1 parent 82ce7a3
commit 22f02c6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/modules/docker-compose.simulation.yaml b/modules/docker-compose.simulation.yaml
@@ -5,11 +5,16 @@ services:
     image: carlasim/carla:0.9.13
     environment:
       - DISPLAY=1
-      - CUDA_VISIBLE_DEVICES=0,1,2
-      - NVIDIA_VISIBLE_DEVICES=0,1,2
-    runtime: nvidia
+      - CUDA_VISIBLE_DEVICES=0
+      - NVIDIA_VISIBLE_DEVICES=0
     restart: always
     command: /bin/bash -c "./CarlaUE4.sh -nosound -carla-server -RenderOffscreen -world-port=2000 -quality-level=Low"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              capabilities: [gpu]
 
   carla_ros_bridge:
     build:

diff --git a/src/samples/python/aggregator/setup.py b/src/samples/python/aggregator/setup.py
@@ -14,7 +14,7 @@
         # Include our package.xml file
         (os.path.join('share', package_name), ['package.xml']),
         # Include all launch files.
-        (os.path.join('share', package_name, 'launch'), \
+        (os.path.join('share', package_name, 'launch'),
          glob(os.path.join('launch', '*.launch.py'))),
     ],
     install_requires=['setuptools'],