diff --git a/pkgs/development/cuda-modules/generic-builders/multiplex.nix b/pkgs/development/cuda-modules/generic-builders/multiplex.nix index ce50653bb5d4..15dd0c938f62 100644 --- a/pkgs/development/cuda-modules/generic-builders/multiplex.nix +++ b/pkgs/development/cuda-modules/generic-builders/multiplex.nix @@ -81,7 +81,7 @@ let redistArch = flags.getRedistArch hostPlatform.system; preferable = - p1: p2: (isSupported p2 -> isSupported p1) && (strings.versionAtLeast p1.version p2.version); + p1: p2: (isSupported p2 -> isSupported p1) && (strings.versionOlder p2.version p1.version); # All the supported packages we can build for our platform. # perSystemReleases :: List Package 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}