build: Just set torch>=2.0 (#1210)

Summary: This project is in maintenance mode so set it free from having a hard dependency on a single version of PyTorch Pull Request resolved: #1210 Reviewed By: abhinavarora Differential Revision: D50347787 Pulled By: seemethere fbshipit-source-id: bb62a8848cb506e21a6024134071e2cf7f729388
pytorch · Oct 31, 2023 · 6d4cdf7 · 6d4cdf7
1 parent c5f2204
commit 6d4cdf7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/.github/workflows/build_conda_linux.yml b/.github/workflows/build_conda_linux.yml
@@ -16,6 +16,7 @@ jobs:
       test-infra-repository: pytorch/test-infra
       test-infra-ref: main
       with-cuda: disable
+      with-rocm: disable
   build:
     needs: generate-matrix
     strategy:

diff --git a/.github/workflows/build_wheels_linux.yml b/.github/workflows/build_wheels_linux.yml
@@ -16,6 +16,7 @@ jobs:
       test-infra-repository: pytorch/test-infra
       test-infra-ref: main
       with-cuda: disable
+      with-rocm: disable
   build:
     needs: generate-matrix
     strategy:

diff --git a/setup.py b/setup.py
@@ -110,11 +110,7 @@ def _get_requirements():
 
 
 # Use new version of torch on main branch
-pytorch_package_dep = "torch>2.0"
-if os.getenv("PYTORCH_VERSION"):
-    pytorch_package_dep = pytorch_package_dep.split(">")[0]
-    pytorch_package_dep += "==" + os.getenv("PYTORCH_VERSION")
-
+pytorch_package_dep = "torch>=2"
 
 requirements = _get_requirements()
 requirements.append(pytorch_package_dep)