diff --git a/docker-compose.yml b/docker-compose.yml
index e0f8229..63b3a9f 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,7 +11,7 @@ x-base_service: &base_service
       resources:
         reservations:
           devices:
-              - driver: nvidia
+              - driver: Nivida1050
                 device_ids: ['0']
                 capabilities: [gpu]