diff --git a/home/Docker.nix b/home/Docker.nix
index 29de8da..53723a9 100644
--- a/home/Docker.nix
+++ b/home/Docker.nix
@@ -8,7 +8,7 @@
   ];
 
   home.packages = with pkgs; [
-    micromamba
+    python312Packages.pytorchWithCuda
   ];
 
   home.sessionVariables = {