cudaPackages: 11.8 -> 12.2

2024-01-17 00:49:27 +00:00
parent 7d161de781
commit 1620cc6e9a
1 changed files with 1 additions and 3 deletions
@@ -7324,9 +7324,7 @@ with pkgs;
  # compatibility: https://www.tensorflow.org/install/source#gpu
  cudaPackagesGoogle = cudaPackages_11;

-  # TODO: try upgrading once there is a cuDNN release supporting CUDA 12. No
-  # such cuDNN release as of 2023-01-10.
-  cudaPackages = cudaPackages_11;
+  cudaPackages = recurseIntoAttrs cudaPackages_12;

  # TODO: move to alias
  cudatoolkit = cudaPackages.cudatoolkit;