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

Merge pull request #169671 from SomeoneSerge/tf-gcc1

Browse Source 
python3Packages.tensorflow*: unpin cuda and fix for gcc11
This commit is contained in:
Samuel Ainsworth
2022-04-22 08:47:52 -07:00
committed by GitHub
parent 02c56def97 5885de0b49
commit 3f47237932
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
@@ -9964,15 +9964,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 {