From 1c92ae96141390907d63a6a5a315b9d040cbcf17 Mon Sep 17 00:00:00 2001 From: Martin Schwaighofer Date: Sun, 24 Mar 2024 21:59:09 +0100 Subject: [PATCH] python3Packages.torchWithRocm: switch back to ROCm 5.7 --- pkgs/development/python-modules/torch/default.nix | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix index 10eecd1de99b7c..82ad61a191674e 100644 --- a/pkgs/development/python-modules/torch/default.nix +++ b/pkgs/development/python-modules/torch/default.nix @@ -52,7 +52,7 @@ # ROCm dependencies rocmSupport ? config.rocmSupport, - rocmPackages, + rocmPackages_5, gpuTargets ? [ ] }: @@ -60,6 +60,8 @@ let inherit (lib) attrsets lists strings trivial; inherit (cudaPackages) cudaFlags cudnn nccl; + rocmPackages = rocmPackages_5; + setBool = v: if v then "1" else "0"; # https://github.com/pytorch/pytorch/blob/v2.0.1/torch/utils/cpp_extension.py#L1744