Update PyTorch requirement to >=2.0

.conda/meta.yaml

@@ -17,7 +17,7 @@ requirements:
 
   run:
     - python>=3.8
-    - pytorch>=1.11
+    - pytorch>=2.0
     - scikit-learn
     - jaxtyping>=0.2.9
     - linear_operator>=0.5.2

.github/workflows/run_test_suite.yml

@@ -50,7 +50,7 @@ jobs:
         if [[ ${{ matrix.pytorch-version }} = "master" ]]; then
           pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html;
         else
-          pip install torch==1.11+cpu -f https://download.pytorch.org/whl/torch_stable.html;
+          pip install torch==2.0.1 --index-url https://download.pytorch.org/whl/cpu
         fi
         pip install -e .
         if [[ ${{ matrix.extras }} == "with-extras" ]]; then

README.md

@@ -30,7 +30,7 @@ See our [**documentation, examples, tutorials**](https://gpytorch.readthedocs.io
 
 **Requirements**:
 - Python >= 3.8
-- PyTorch >= 1.11
+- PyTorch >= 2.0
 
 Install GPyTorch using pip or conda:
 

setup.py

@@ -37,7 +37,7 @@ def find_version(*file_paths):
 readme = open("README.md").read()
 
 
-torch_min = "1.11"
+torch_min = "2.0"
 install_requires = [
     "jaxtyping>=0.2.9",
     "mpmath>=0.19,<=1.3",  # avoid incompatibiltiy with torch+sympy with mpmath 1.4