diff --git a/setup.py b/setup.py index 96a100c1b..334bd1457 100644 --- a/setup.py +++ b/setup.py @@ -65,9 +65,7 @@ def write_version_file(version): def _get_pytorch_version(is_nightly): # if "PYTORCH_VERSION" in os.environ: # return f"torch=={os.environ['PYTORCH_VERSION']}" - if is_nightly: - return "torch>=2.3.0.dev" - return "torch>=2.2.1" + return "torch==2.2.2" def _get_packages():