Require PyTorch>=2.0.1, GPyTorch==1.13 and linear_operator==0.5.3

.conda/meta.yaml

@@ -17,9 +17,9 @@ requirements:
     - setuptools
     - setuptools_scm
   run:
-    - pytorch >=1.13.1
-    - gpytorch ==1.12
-    - linear_operator ==0.5.2
+    - pytorch >=2.0
+    - gpytorch ==1.12.1
+    - linear_operator ==0.5.3
     - scipy
     - multipledispatch
     - pyro-ppl >=1.8.4

.github/workflows/test_stable.yml

@@ -62,8 +62,6 @@ jobs:
         python setup.py egg_info
         req_txt="botorch.egg-info/requires.txt"
         min_torch_version=$(grep '\btorch[>=]=' ${req_txt} | sed 's/[^0-9.]//g')
-        # HACK around the fact that pytorch does not offer a mac binary for 1.13.1 for py3.11 - TODO: Remove when bumping torch to 2.0.1
-        min_torch_version="$(if ${{ matrix.python-version == '3.11' }} && ${{ matrix.os == 'macos-14' }}; then echo "2.0.1"; else echo "${min_torch_version}"; fi)"
         min_gpytorch_version=$(grep '\bgpytorch[>=]=' ${req_txt} | sed 's/[^0-9.]//g')
         min_linear_operator_version=$(grep '\blinear_operator[>=]=' ${req_txt} | sed 's/[^0-9.]//g')
         pip install "torch==${min_torch_version}" "gpytorch==${min_gpytorch_version}" "linear_operator==${min_linear_operator_version}"

README.md

@@ -56,9 +56,9 @@ Optimization simply use Ax.
 
 **Installation Requirements**
 - Python >= 3.10
-- PyTorch >= 1.13.1
-- gpytorch == 1.12
-- linear_operator == 0.5.2
+- PyTorch >= 2.0
+- gpytorch == 1.12.1
+- linear_operator == 0.5.3
 - pyro-ppl >= 1.8.4
 - scipy
 - multiple-dispatch

environment.yml

@@ -4,9 +4,9 @@ channels:
   - gpytorch
   - conda-forge
 dependencies:
-  - pytorch>=1.13.1
-  - gpytorch==1.12
-  - linear_operator==0.5.2
+  - pytorch>=2.0
+  - gpytorch==1.12.1
+  - linear_operator==0.5.3
   - scipy
   - multipledispatch
   - pyro-ppl>=1.8.4

requirements.txt

@@ -1,7 +1,7 @@
 multipledispatch
 scipy
 mpmath>=0.19,<=1.3
-torch>=1.13.1
+torch>=2.0
 pyro-ppl>=1.8.4
-gpytorch==1.12
-linear_operator==0.5.2
+gpytorch==1.12.1
+linear_operator==0.5.3