From bbaad9c2fa76d8bd884f25bca14b6dcdb9cc12bd Mon Sep 17 00:00:00 2001 From: Samuel Ainsworth Date: Thu, 20 Oct 2022 14:18:08 -0700 Subject: [PATCH] python3Packages.jaxlib{-bin}: no longer need to pin the cudaPackages version --- pkgs/top-level/python-packages.nix | 6 ------ 1 file changed, 6 deletions(-) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 4509e33aef52..5a100604dacd 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -4780,9 +4780,6 @@ in { jaxlib-bin = callPackage ../development/python-modules/jaxlib/bin.nix { cudaSupport = pkgs.config.cudaSupport or false; - # At the time of writing (2022-04-18), `cudaPackages.nccl` is broken, so we - # pin to `cudaPackages_11_6` instead. - cudaPackages = pkgs.cudaPackages_11_6; }; jaxlib-build = callPackage ../development/python-modules/jaxlib rec { @@ -4792,9 +4789,6 @@ in { }; # Some platforms don't have `cudaSupport` defined, hence the need for 'or false'. cudaSupport = pkgs.config.cudaSupport or false; - # At the time of writing (2022-04-18), `cudaPackages.nccl` is broken, so we - # pin to `cudaPackages_11_6` instead. - cudaPackages = pkgs.cudaPackages_11_6; IOKit = pkgs.darwin.apple_sdk_11_0.IOKit; protobuf = pkgs.protobuf3_20; # jaxlib-build 0.3.15 won't build with protobuf 3.21 };