python3Packages.torch: Fix 'setuptools' not being found with ROCm build

2023-10-10 01:43:09 -05:00 · 2023-10-10 01:43:09 -05:00 · aeccee810a
commit aeccee810a
parent 58f70713eb
1 changed files with 5 additions and 0 deletions
--- a/pkgs/development/python-modules/torch/default.nix
+++ b/pkgs/development/python-modules/torch/default.nix
@ -101,6 +101,11 @@ let
      rocminfo rocm-thunk rocm-comgr rocm-device-libs
      rocm-runtime clr.icd hipify
    ];
+
+    # Fix `setuptools` not being found
+    postBuild = ''
+      rm -rf $out/nix-support
+    '';
  };

  brokenConditions = attrsets.filterAttrs (_: cond: cond) {