From 8fa343c6b4451df0ad9ecb976e76e793ccb8f90a Mon Sep 17 00:00:00 2001 From: Sergei Trofimovich Date: Fri, 18 Apr 2025 22:58:45 +0100 Subject: [PATCH] python3Packages.torch: fix cudnn attribute access Without the change the eval fails as: $ nix build --no-link -f. opensplatWithCuda --argstr system aarch64-linux error: error: attribute 'lib' missing at /home/slyfox/dev/git/nixpkgs-master/pkgs/development/python-modules/torch/default.nix:357:30: 356| export CUDNN_INCLUDE_DIR=${lib.getLib cudnn}/include 357| export CUDNN_LIB_DIR=${cudnn.lib}/lib | ^ 358| '' --- pkgs/development/python-modules/torch/source/default.nix | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix index 3fd0e9d11ee7..187d7821b2cd 100644 --- a/pkgs/development/python-modules/torch/source/default.nix +++ b/pkgs/development/python-modules/torch/source/default.nix @@ -364,7 +364,7 @@ buildPythonPackage rec { '' + lib.optionalString (cudaSupport && cudaPackages ? cudnn) '' export CUDNN_INCLUDE_DIR=${lib.getLib cudnn}/include - export CUDNN_LIB_DIR=${cudnn.lib}/lib + export CUDNN_LIB_DIR=${lib.getLib cudnn}/lib '' + lib.optionalString rocmSupport '' export ROCM_PATH=${rocmtoolkit_joined}