Import the correct onnxruntime.

Author: Narsil

flake.lock

@@ -148,6 +148,11 @@
         #       hash = "sha256-1AN2E9t/lZhbXdVznhTcniy+7ZzlaEp/gwLEAucs6EA=";
         #       # hash = lib.fakeHash;
         #     };
+        mkl2024 = import ./nix/mkl.nix;
+
+        onnxruntimeGcc13 = pkgs.onnxruntime.override {
+          stdenv = pkgs.cudaPackages.backendStdenv;
+        };
 
       in
       #     cargoDeps = pkgs.rustPlatform.fetchCargoVendor {
@@ -195,7 +200,7 @@
         devShells.default =
           pkgs.mkShell.override
             {
-              stdenv = pkgs.gcc13Stdenv;
+              stdenv = pkgs.cudaPackages.backendStdenv;
             }
             {
 
@@ -208,6 +213,8 @@
                 cudaPackages.cudatoolkit
                 python3Packages.python
                 python3Packages.venvShellHook
+                onnxruntimeGcc13
+                mkl
               ];
               venvDir = "./.venv";
               LD_LIBRARY_PATH = "${pkgs.stdenv.cc.cc.lib}/lib:/run/opengl-driver/lib";