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 {