gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python3Packages.paddlepaddle: fix evaluation error when selecting unsupported cuda version

Browse Source
This commit is contained in:
Gaetan Lepage
2026-01-10 09:52:24 +00:00
parent 9e2d8102de
commit 5a9f5ec038
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/paddlepaddle/default.nix
+16 -12
View File
@@ -34,21 +34,25 @@ let
version = sources.version;
format = "wheel";
pyShortVersion = "cp${builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion}";
hash =
if cudaSupport then
sources."${stdenv.hostPlatform.system}"."gpu"."cu${
lib.replaceStrings [ "." ] [ "" ] cudaPackages.cudatoolkit.version
}"."${pyShortVersion}"
else
sources."${stdenv.hostPlatform.system}"."cpu"."${pyShortVersion}"
or (throw "Unsupported system: ${stdenv.hostPlatform.system}");
platform = sources."${stdenv.hostPlatform.system}".platform;
cudaVersion = "cu${lib.replaceStrings [ "." ] [ "" ] cudaPackages.cudatoolkit.version}";
throwSystem = throw "Unsupported system: ${stdenv.hostPlatform.system}";
systemSources = sources."${stdenv.hostPlatform.system}" or throwSystem;
supportedCudaVersions = lib.concatStringsSep ", " (builtins.attrNames systemSources.gpu);
throwCuda = throw "Unsupported CUDA version: ${cudaVersion}. Supported versions: ${supportedCudaVersions}";
platformSources =
if cudaSupport then systemSources.gpu.${cudaVersion} or throwCuda else systemSources.cpu;
throwPython = throw "Unsupported python version: ${pyShortVersion}";
hash = platformSources.${pyShortVersion} or throwPython;
platform = sources.${stdenv.hostPlatform.system}.platform;
src =
if cudaSupport then
(fetchurl {
url = "https://paddle-whl.bj.bcebos.com/stable/cu${
lib.replaceStrings [ "." ] [ "" ] cudaPackages.cudatoolkit.version
}/paddlepaddle-gpu/paddlepaddle_gpu-${version}-${pyShortVersion}-${pyShortVersion}-linux_x86_64.whl";
url = "https://paddle-whl.bj.bcebos.com/stable/${cudaVersion}/paddlepaddle-gpu/paddlepaddle_gpu-${version}-${pyShortVersion}-${pyShortVersion}-linux_x86_64.whl";
inherit hash;
})
else