diff --git a/docker/docker-compose-gui-nvidia.yml b/docker/docker-compose-gui-nvidia.yml
index 76793e3..b7c5a44 100644
--- a/docker/docker-compose-gui-nvidia.yml
+++ b/docker/docker-compose-gui-nvidia.yml
@@ -6,4 +6,5 @@ services:
       service: stereo_matching_docker
     environment:
      - NVIDIA_VISIBLE_DEVICES=all
+     - NVIDIA_DRIVER_CAPABILITIES=all
     runtime: nvidia