gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python3Packages.tensorflow*: unpin cuda

Browse Source
This commit is contained in:
Someone Serge
2022-04-21 13:24:59 +03:00
parent 92c049659a
commit 5885de0b49
1 changed files with 1 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/top-level/python-packages.nix
+1 -8
View File
@@ -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 {