nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-09-29 11:20:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.paddlepaddle: fix build with cudaSupport

Browse Source 
When `cudaSupport = true`, the GPU wheel declares PyPI `nvidia-*-cu12` packages as dependencies. nixpkgs provides CUDA via system packages, so `pythonRuntimeDepsCheckHook` fails. Remove the 16 PyPI nvidia/cuda packages from the wheel metadata with `pythonRemoveDeps`, conditional on `cudaSupport`.

Assisted-by: OpenCode(1.18.4+411eff7)/ses_02881e43effeU5rUbrCeB05McK:zai-coding-plan/glm-5.2
This commit is contained in:
Daniel Fahey
2026-08-06 23:26:56 +01:00
parent 70a9c9eeda
commit 6ff190260e
1 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
pkgs/development/python-modules/paddlepaddle/default.nix
View File

@@ -77,6 +77,25 @@ buildPythonPackage {
"opt_einsum"
];
pythonRemoveDeps = lib.optionals cudaSupport [
"cuda-python"
"nvidia-cublas-cu12"
"nvidia-cuda-cccl-cu12"
"nvidia-cuda-cupti-cu12"
"nvidia-cuda-nvrtc-cu12"
"nvidia-cuda-runtime-cu12"
"nvidia-cudnn-cu12"
"nvidia-cufile-cu12"
"nvidia-cufft-cu12"
"nvidia-curand-cu12"
"nvidia-cusolver-cu12"
"nvidia-cusparse-cu12"
"nvidia-cusparselt-cu12"
"nvidia-nccl-cu12"
"nvidia-nvjitlink-cu12"
"nvidia-nvtx-cu12"
];
dependencies = [
setuptools
httpx