diff options
author | Connor Baker <connor.baker@tweag.io> | 2023-11-07 14:35:11 +0000 |
---|---|---|
committer | Connor Baker <connor.baker@tweag.io> | 2023-12-07 16:45:54 +0000 |
commit | 8e800cedaf24f5ad9717463b809b0beef7677000 (patch) | |
tree | 6cd091e8af79bf9d036b76f87105047143b620b9 /pkgs/development/cuda-modules/modules/tensorrt/default.nix | |
parent | 397d95d07fd095a3fba459a694bd284be3c47899 (diff) |
cudaPackages: move derivations to cuda-modules & support aarch64
cudaPackages.cuda_compat: ignore missing libs provided at runtime cudaPackages.gpus: Jetson should never build by default cudaPackages.flags: don't build Jetson capabilities by default cudaPackages: re-introduce filter for pre-existing CUDA redist packages in overrides cudaPackages: only recurseIntoAttrs for the latest of each major version cudaPackages.nvccCompatabilities: use GCC 10 through CUDA 11.5 to avoid a GLIBC incompatability cudaPackages.cutensor: acquire libcublas through cudatoolkit prior to 11.4 cudaPackages.cuda_compat: mark as broken on aarch64-linux if not targeting Jetson cudaPackages.cutensor_1_4: fix build cudaPackages: adjust use of autoPatchelfIgnoreMissingDeps cudaPackages.cuda_nvprof: remove unecessary override to add addOpenGLRunpath cudaPackages: use getExe' to avoid patchelf warning about missing meta.mainProgram cudaPackages: fix evaluation with Nix 2.3 cudaPackages: fix platform detection for Jetson/non-Jetson aarch64-linux python3Packages.tensorrt: mark as broken if required packages are missing Note: evaluating the name of the derivation will fail if tensorrt is not present, which is why we wrap the value in `lib.optionalString`. cudaPackages.flags.getNixSystem: add guard based on jetsonTargets cudaPackages.cudnn: use explicit path to patchelf cudaPackages.tensorrt: use explicit path to patchelf
Diffstat (limited to 'pkgs/development/cuda-modules/modules/tensorrt/default.nix')
-rw-r--r-- | pkgs/development/cuda-modules/modules/tensorrt/default.nix | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/pkgs/development/cuda-modules/modules/tensorrt/default.nix b/pkgs/development/cuda-modules/modules/tensorrt/default.nix new file mode 100644 index 0000000000000..e62942c679aa0 --- /dev/null +++ b/pkgs/development/cuda-modules/modules/tensorrt/default.nix @@ -0,0 +1,16 @@ +{options, ...}: +{ + options.tensorrt.releases = options.generic.releases; + # TODO(@connorbaker): Figure out how to add additional options to the + # to the generic release. + # { + # cudnnVersion = lib.options.mkOption { + # description = "The CUDNN version supported"; + # type = types.nullOr majorMinorVersion; + # }; + # filename = lib.options.mkOption { + # description = "The tarball name"; + # type = types.str; + # }; + # } +} |