Merge pull request #210135 from samuela/samuela/cuda12

cudaPackages_12: init at 12.0.0
This commit is contained in:
Samuel Ainsworth 2023-01-16 13:47:38 -05:00 committed by GitHub
commit 86f29625e7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 1144 additions and 0 deletions

View file

@ -13,6 +13,7 @@ final: prev: let
"11.6" = ./manifests/redistrib_11.6.2.json;
"11.7" = ./manifests/redistrib_11.7.0.json;
"11.8" = ./manifests/redistrib_11.8.0.json;
"12.0" = ./manifests/redistrib_12.0.0.json;
};
# Function to build a single cudatoolkit redist package

File diff suppressed because it is too large Load diff

View file

@ -71,3 +71,13 @@ version = "11.8.0"
url = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
sha256 = "sha256-kiPErzrr5Ke77Zq9mxY7A6GzS4VfvCtKDRtwasCaWhY="
gcc = "gcc11"
["12.0"]
version = "12.0.0"
url = "https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run"
sha256 = "sha256-kF6blRaQCDn7dgZHGdt1JDnzi4y3MLSTNdi9U93605I="
# CUDA 12 is compatible with gcc12, but nixpkgs default gcc is still on gcc11 as
# of 2023-01-08. See https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#system-requirements.
# This should be upgraded to gcc12 once nixpkgs default gcc is upgraded. Other
# CUDA versions should likely have their gcc versions upgraded as well.
gcc = "gcc11"

View file

@ -6223,6 +6223,12 @@ with pkgs;
cudaPackages_11_7 = callPackage ./cuda-packages.nix { cudaVersion = "11.7"; };
cudaPackages_11_8 = callPackage ./cuda-packages.nix { cudaVersion = "11.8"; };
cudaPackages_11 = cudaPackages_11_7;
cudaPackages_12_0 = callPackage ./cuda-packages.nix { cudaVersion = "12.0"; };
cudaPackages_12 = cudaPackages_12_0;
# TODO: try upgrading once there is a cuDNN release supporting CUDA 12. No
# such cuDNN release as of 2023-01-10.
cudaPackages = recurseIntoAttrs cudaPackages_11;
# TODO: move to alias