python3Packages.tensorflow*: unpin cuda

main
Someone Serge 2 years ago
parent 92c049659a
commit 5885de0b49
No known key found for this signature in database
GPG Key ID: 7B0E3B1390D61DA4
  1. 9
      pkgs/top-level/python-packages.nix

@ -9956,15 +9956,8 @@ in {
tensorboardx = callPackage ../development/python-modules/tensorboardx { };
tensorflow-bin = let
# CUDA-related packages that are compatible with the currently packaged version
# of TensorFlow, used to keep these versions in sync in related packages like `jaxlib`.
cudaPackages = pkgs.cudaPackages_11_2.overrideScope' (final: prev: {
cudnn = prev.cudnn_8_1_1;
});
in callPackage ../development/python-modules/tensorflow/bin.nix {
tensorflow-bin = callPackage ../development/python-modules/tensorflow/bin.nix {
cudaSupport = pkgs.config.cudaSupport or false;
inherit cudaPackages;
};
tensorflow-build = callPackage ../development/python-modules/tensorflow {

Loading…
Cancel
Save