Merge pull request #1582 from Trusted-AI/dependabot/pip/torch-1.11.0

Bump torch from 1.10.2+cpu to 1.11.0

Author: beat-buesser

.github/workflows/ci-lingvo.yml

@@ -47,7 +47,7 @@ jobs:
           sudo apt-get update
           sudo apt-get -y -q install ffmpeg libavcodec-extra
           python -m pip install --upgrade pip setuptools wheel
-          pip install -q -r <(sed '/^scipy/d;/^matplotlib/d;/^pandas/d;/^statsmodels/d;/^numba/d;/^jax/d;/^h5py/d;/^Pillow/d;/^pytest-mock/d' requirements_test.txt)
+          pip install -q -r <(sed '/^scipy/d;/^matplotlib/d;/^pandas/d;/^statsmodels/d;/^numba/d;/^jax/d;/^h5py/d;/^Pillow/d;/^pytest-mock/d;/^torch/d;/^torchaudio/d;/^torchvision/d' requirements_test.txt)
           pip install scipy==1.5.4
           pip install matplotlib==3.3.4
           pip install pandas==1.1.5
@@ -61,6 +61,9 @@ jobs:
           pip install jax[cpu]==0.2.17
           pip install h5py==2.10.0
           pip install pytest-mock
+          pip install torch==1.10.2+cpu --find-links https://download.pytorch.org/whl/cpu/torch_stable.html
+          pip install torchaudio==0.10.2+cpu --find-links https://download.pytorch.org/whl/cpu/torch_stable.html
+          pip install torchvision==0.11.3+cpu --find-links https://download.pytorch.org/whl/cpu/torch_stable.html
           pip list
       - name: Run ${{ matrix.name }} Tests
         run: ./run_tests.sh ${{ matrix.framework }}

.github/workflows/ci-pytorch.yml

@@ -25,12 +25,6 @@ jobs:
       fail-fast: false
       matrix:
         include:
-          - name: PyTorch 1.8.1 (Python 3.8)
-            framework: pytorch
-            python: 3.8
-            torch: 1.8.1+cpu
-            torchvision: 0.9.1+cpu
-            torchaudio: 0.8.1
           - name: PyTorch 1.9.1 (Python 3.8)
             framework: pytorch
             python: 3.8
@@ -43,6 +37,12 @@ jobs:
             torch: 1.10.2+cpu
             torchvision: 0.11.3+cpu
             torchaudio: 0.10.2+cpu
+          - name: PyTorch 1.11.0 (Python 3.8)
+            framework: pytorch
+            python: 3.8
+            torch: 1.11.0+cpu
+            torchvision: 0.12.0+cpu
+            torchaudio: 0.11.0
 
     name: ${{ matrix.name }}
     steps:

requirements_test.txt

@@ -30,9 +30,9 @@ mxnet-native==1.8.0.post0
 
 # PyTorch
 --find-links https://download.pytorch.org/whl/cpu/torch_stable.html
-torch==1.10.2+cpu
-torchaudio==0.10.2+cpu
-torchvision==0.11.3+cpu
+torch==1.11.0
+torchaudio==0.11.0+cpu
+torchvision==0.12.0+cpu
 
 catboost==1.0.4
 GPy==1.10.0