diff --git a/system/shen/configuration.nix b/system/shen/configuration.nix
index 09ef0ab..9bb600f 100644
--- a/system/shen/configuration.nix
+++ b/system/shen/configuration.nix
@@ -54,9 +54,9 @@
   };
 
   # environment.variables.AMD_VULKAN_ICD = lib.mkDefault "RADV";
-  # environment.systemPackages = with pkgs; [
-  #   nvidia-docker
-  # ];
+  environment.systemPackages = with pkgs; [
+    cudatoolkit
+  ];
 
   virtualisation.docker.enableNvidia = true;