From 0f76efb48143e4f38f7714588e3cb38d055404b2 Mon Sep 17 00:00:00 2001 From: Someone Serge Date: Tue, 21 Mar 2023 16:30:06 +0200 Subject: [PATCH] python3Packages.torchWithRocm: ignore config.cudaSupport --- pkgs/top-level/python-packages.nix | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 1e9075ace6ad..5349c922a4e9 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -11856,6 +11856,7 @@ self: super: with self; { torchWithCuda = self.torch.override { magma = pkgs.magma-cuda; cudaSupport = true; + rocmSupport = false; }; torchWithoutCuda = self.torch.override { @@ -11865,6 +11866,7 @@ self: super: with self; { torchWithRocm = self.torch.override { magma = pkgs.magma-hip; rocmSupport = true; + cudaSupport = false; }; torchWithoutRocm = self.torch.override {