diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 25b8065..7b027dd 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -8,7 +8,7 @@ services:
         - username=devel
         - userid=1000
     container_name: handson-ml2
-    image: handson-ml2
+    image: ageron/handson-ml2:latest #latest-gpu
     restart: unless-stopped
     logging:
       driver: json-file