diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix index 56396da3f78d..33ada3baab2c 100644 --- a/pkgs/development/python-modules/torch/source/default.nix +++ b/pkgs/development/python-modules/torch/source/default.nix @@ -1,6 +1,7 @@ { stdenv, lib, + pkgs, fetchFromGitHub, fetchFromGitLab, fetchpatch2, @@ -250,6 +251,9 @@ let "Magma cudaPackages does not match cudaPackages" = cudaSupport && (effectiveMagma.cudaPackages.cudaMajorMinorVersion != cudaPackages.cudaMajorMinorVersion); + # Should be fixed by WIP https://github.com/NixOS/nixpkgs/pull/438399 + "ROCm support on non-default python versions is temporarily broken" = + rocmSupport && (pkgs.python3.version != python.version); }; unroll-src = writeShellScript "unroll-src" ''