libtorch-bin: move cudaSupport logic to package

Follow the same pattern as the majority of packages using cuda
This commit is contained in:
zimbatm
2024-07-09 14:57:31 +02:00
parent f571ea25a6
commit da07f98d58
2 changed files with 3 additions and 4 deletions
@@ -4,12 +4,13 @@
, lib
, libcxx
, llvmPackages
, config
, addOpenGLRunpath
, patchelf
, fixDarwinDylibNames
, cudaSupport
, cudaSupport ? config.cudaSupport
}:
let
+1 -3
View File
@@ -4256,9 +4256,7 @@ with pkgs;
libtensorflow = python3.pkgs.tensorflow.libtensorflow;
libtorch-bin = callPackage ../development/libraries/science/math/libtorch/bin.nix {
inherit (config) cudaSupport;
};
libtorch-bin = callPackage ../development/libraries/science/math/libtorch/bin.nix { };
tensorflow-lite = callPackage ../development/libraries/science/math/tensorflow-lite { };