Change package version (TensorRT)

Help
9

yeah i got below shell.nix working on my vps

{
  pkgs ? import <nixpkgs> { },
}:
let
  tensorrt_10_7 = pkgs.cudaPackages_11.tensorrt_10_8.overrideAttrs (_: {
    version = "10.7.0.23";
    # wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.7.0/tars/TensorRT-10.7.0.23.Linux.x86_64-gnu.cuda-12.6.tar.gz
    src = pkgs.requireFile {
      name = "TensorRT-10.7.0.23.Linux.x86_64-gnu.cuda-12.6.tar.gz";
      # hash = pkgs.lib.fakeHash;
      hash = "sha256-1/FlIEV8qvl62KfpTYAvidd67fnzYaJV8sIW4qOkChE="; # nix hash file TensorRT-10.7.0.23.Linux.x86_64-gnu.cuda-12.6.tar.gz
      message = ''
        1. wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.7.0/tars/TensorRT-10.7.0.23.Linux.x86_64-gnu.cuda-12.6.tar.gz
        2. nix-store --add-fixed sha256 TensorRT-10.7.0.23.Linux.x86_64-gnu.cuda-12.6.tar.gz
        3. nix hash file TensorRT-10.7.0.23.Linux.x86_64-gnu.cuda-12.6.tar.gz
      '';
    };
  });
in
pkgs.mkShell {
  name = "cuda-env-shell";
  buildInputs = [ tensorrt_10_7 ];
  shellHook = ''
    export LD_LIBRARY_PATH=${tensorrt_10_7.lib}/lib
  '';
}

sorry for my unprofessional guidance :sob: (also thanks @SergeK for correction)
took me a while to understand the pile of code inside cuda module

2 Likes