diff --git a/constraints/rocm.txt b/constraints/rocm.txt index 027833533..8745ba6c4 100644 --- a/constraints/rocm.txt +++ b/constraints/rocm.txt @@ -1 +1 @@ ---extra-index-url https://download.pytorch.org/whl/rocm5.4.2/ +--extra-index-url https://download.pytorch.org/whl/nightly/rocm6.0