add cudasupport

flake.nix

@@ -22,6 +22,10 @@
         pkgs = import inputs.nixpkgs {
           inherit system;
           overlays = [ inputs.rust-overlay.overlays.default ];
+            config = {
+              allowUnfree = true;
+              cudaSupport = true;
+            };
         };
         lib = pkgs.lib;
 
@@ -99,6 +103,9 @@
               (lib.python_package ppkgs)
             ]))
           ];
+          shellHook = ''
+            export CUDA_PATH=${pkgs.cudatoolkit}
+          '';
         };
         lib = {
           # To use in other builds with the "withPackages" call