Add gpu drivers for count

hosts/count/configuration.nix

@@ -24,6 +24,27 @@
   # Use latest kernel.
   boot.kernelPackages = pkgs.linuxPackages_latest;
 
+# Enable AMD GPU drivers
+	boot.initrd.kernelModules = [ "amdgpu" ];
+	services.xserver.videoDrivers = [ "amdgpu" ];
+
+	# Enable ROCm support
+	hardware.graphics.enable = true;
+	hardware.amdgpu.opencl.enable = true;
+
+	# Add the ROCm packages to your system
+	environment.systemPackages = with pkgs; [
+		rocmPackages.rocminfo
+		rocmPackages.rocm-smi
+	];
+
+	services.ollama = {
+		enable = true;
+		acceleration = "rocm";
+		# Force the RX 6600 to use compatible ROCm kernels
+		rocmOverrideGfx = "10.3.0";
+	};
+
   # Setup keyfile
   boot.initrd.secrets = {
     "/boot/crypto_keyfile.bin" = null;