diff --git a/setup.py b/setup.py index ebc34985c..7b0221c25 100644 --- a/setup.py +++ b/setup.py @@ -18,7 +18,7 @@ def get_default_dependencies(): ] elif platform == "rocm": return [ - "torch>=2.6.0.dev", + "torch>=2.6.0.dev20241122+rocm6.2", "triton>=3.0.0", ]