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

python3Packages.torch: fix cudnn attribute access

Browse Source 
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|     ''
This commit is contained in:
Sergei Trofimovich
2025-05-08 17:20:38 +00:00
committed by SomeoneSerge
parent 2ba0dcad2e
commit 8fa343c6b4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/torch/source/default.nix
+1 -1
View File
@@ -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}