diff --git a/pkgs/development/python-modules/paddlepaddle/default.nix b/pkgs/development/python-modules/paddlepaddle/default.nix index 05342e2fb68e..250ec8a11d89 100644 --- a/pkgs/development/python-modules/paddlepaddle/default.nix +++ b/pkgs/development/python-modules/paddlepaddle/default.nix @@ -13,7 +13,7 @@ zlib, setuptools, cudaSupport ? config.cudaSupport or false, - cudaPackages, + cudaPackages_12_9, addDriverRunpath, # runtime dependencies httpx, @@ -23,6 +23,7 @@ decorator, astor, opt-einsum, + safetensors, typing-extensions, }: @@ -40,7 +41,7 @@ let src = if cudaSupport then (fetchurl { - url = "https://paddle-whl.bj.bcebos.com/stable/cu128/paddlepaddle-gpu/paddlepaddle-${version}-${pyShortVersion}-${pyShortVersion}-linux_x86_64.whl"; + url = "https://paddle-whl.bj.bcebos.com/stable/cu129/paddlepaddle-gpu/paddlepaddle_gpu-${version}-${pyShortVersion}-${pyShortVersion}-linux_x86_64.whl"; inherit hash; }) else @@ -81,6 +82,7 @@ buildPythonPackage { decorator astor opt-einsum + safetensors typing-extensions ]; @@ -99,7 +101,7 @@ buildPythonPackage { (lib.getLib stdenv.cc.cc) ] ++ lib.optionals cudaSupport ( - with cudaPackages; + with cudaPackages_12_9; [ cudatoolkit.lib cudatoolkit.out @@ -127,6 +129,8 @@ buildPythonPackage { sed -i 's/^INSTALLED_VERSION=.*/INSTALLED_VERSION="${version}"/' $out/bin/paddle ''; + passthru.updateScript = ./update.sh; + meta = { description = "Machine Learning Framework from Industrial Practice"; homepage = "https://github.com/PaddlePaddle/Paddle"; diff --git a/pkgs/development/python-modules/paddlepaddle/sources.nix b/pkgs/development/python-modules/paddlepaddle/sources.nix index 1f2588fd3aaa..13684502a438 100644 --- a/pkgs/development/python-modules/paddlepaddle/sources.nix +++ b/pkgs/development/python-modules/paddlepaddle/sources.nix @@ -1,28 +1,28 @@ { - version = "3.2.0"; + version = "3.2.2"; x86_64-linux = { platform = "manylinux1_x86_64"; cpu = { - cp312 = "sha256-LBf2daJevQZ19wP3uCd36pLbDDYL1Vpcay36rXvD8mA="; - cp313 = "sha256-+irs0GFCQ1D2dwqD3aB5aR9yJGNc/ydurvTj4XRAA50="; + cp312 = "sha256-9rO3PUesyCz1RYlublwxtVp/mibcCssoq8wkBp2Tlt0="; + cp313 = "sha256-xPSFWWpobqcqsN3/yeQRePRWMUqZ1FOPXRgaAVapJ/o="; }; gpu = { - cp312 = "sha256-YqxAWSvjhYOQUCiUPtjC3PdRxFeWGm/be8FxOrpaLZo="; - cp313 = "sha256-TdFx5Ut3iOTRg8USL2eIHyRYv8QT4KrLBPdiPkaK+Nc="; + cp312 = "sha256-t2beVNEfz+JcesM5+i9BVNKKXIsI3zvz1VrTSd/79+c="; + cp313 = "sha256-0thm6aAGpqSLfJdI0MZKuJPn2e4CsNU5S19ohzkukSs="; }; }; aarch64-linux = { platform = "manylinux2014_aarch64"; cpu = { - cp312 = "sha256-TOUUEUr3Zxh1/Ekn6gBa+51dD9d3rrVqEHjSL39Os/s="; - cp313 = "sha256-KqDlAvbdKE7aIcKH0yZFTfomrMqIwjqgrFgdHvmDGMU="; + cp312 = "sha256-pZ2QILoWD6f1rvyzDkpjKTkKY10aC1EfZFgTk3bAP5k="; + cp313 = "sha256-sDVpuhUaxQqnMJVcZd+lcf565x+UV0yU6L/l4JE6fZw="; }; }; aarch64-darwin = { platform = "macosx_11_0_arm64"; cpu = { - cp312 = "sha256-rDNK9y0bDUnUyTPP1OtC1z7C3HpqryJIxihkPUGbbac="; - cp313 = "sha256-OBhOHqf9e/A4g+ph9FYZuXEtksE60foWk06p4NbT6ZE="; + cp312 = "sha256-xi2rDU323il4lggy8AFaFFwjzTeUGNky3dj1RmKdH+Q="; + cp313 = "sha256-9+m8BGnRPZmSXNdmxKbO61q2BDt5DdQVzgdvu+uzkHQ="; }; }; }