From 253e4ff72a460db714683bfa43f12b4c8bfb370c Mon Sep 17 00:00:00 2001 From: Conroy Cheers Date: Tue, 22 Oct 2024 15:31:23 +1100 Subject: [PATCH] python3Packages.cupy: update build toolchain --- pkgs/development/cuda-modules/cudnn/releases.nix | 6 +++--- pkgs/development/python-modules/cupy/default.nix | 4 +++- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/pkgs/development/cuda-modules/cudnn/releases.nix b/pkgs/development/cuda-modules/cudnn/releases.nix index 586bdecf67c0..13b330f9490b 100644 --- a/pkgs/development/cuda-modules/cudnn/releases.nix +++ b/pkgs/development/cuda-modules/cudnn/releases.nix @@ -7,7 +7,7 @@ { version = "8.9.5.30"; minCudaVersion = "12.0"; - maxCudaVersion = "12.2"; + maxCudaVersion = "12.4"; url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-aarch64/cudnn-linux-aarch64-8.9.5.30_cuda12-archive.tar.xz"; hash = "sha256-BJH3sC9VwiB362eL8xTB+RdSS9UHz1tlgjm/mKRyM6E="; } @@ -75,7 +75,7 @@ { version = "8.9.7.29"; minCudaVersion = "12.0"; - maxCudaVersion = "12.2"; + maxCudaVersion = "12.4"; url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-sbsa/cudnn-linux-sbsa-8.9.7.29_cuda12-archive.tar.xz"; hash = "sha256-6Yt8gAEHheXVygHuTOm1sMjHNYfqb4ZIvjTT+NHUe9E="; } @@ -183,7 +183,7 @@ { version = "8.9.7.29"; minCudaVersion = "12.0"; - maxCudaVersion = "12.2"; + maxCudaVersion = "12.4"; url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz"; hash = "sha256-R1MzYlx+QqevPKCy91BqEG4wyTsaoAgc2cE++24h47s="; } diff --git a/pkgs/development/python-modules/cupy/default.nix b/pkgs/development/python-modules/cupy/default.nix index 83cde3be9a9a..525b593f23e5 100644 --- a/pkgs/development/python-modules/cupy/default.nix +++ b/pkgs/development/python-modules/cupy/default.nix @@ -47,6 +47,8 @@ buildPythonPackage rec { disabled = pythonOlder "3.7"; + stdenv = cudaPackages.backendStdenv; + src = fetchFromGitHub { owner = "cupy"; repo = "cupy"; @@ -118,7 +120,7 @@ buildPythonPackage rec { homepage = "https://cupy.chainer.org/"; changelog = "https://github.com/cupy/cupy/releases/tag/v${version}"; license = licenses.mit; - platforms = [ "x86_64-linux" ]; + platforms = [ "x86_64-linux" "aarch64-linux" ]; maintainers = with maintainers; [ hyphon81 ]; }; }