diff --git a/modules/containers/immich.nix b/modules/containers/immich.nix
index 2175217..9fc9c41 100644
--- a/modules/containers/immich.nix
+++ b/modules/containers/immich.nix
@@ -54,11 +54,15 @@ in
         immich-machine-learning = {
           autoStart = true;
 
-          image = "ghcr.io/immich-app/immich-machine-learning:v2";
+          image = "ghcr.io/immich-app/immich-machine-learning:v2-cuda";
 
           volumes = [
             "${workDir}/cache:/cache"
           ];
+
+          extraOptions = [
+            "--device=nvidia.com/gpu=all"
+          ];
         };
 
         immich-redis = {