Set torch* version for Lingvo workflow

Signed-off-by: Beat Buesser <[email protected]>

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
+          pip install torchaudio==0.10.2+cpu
+          pip install torchvision==0.11.3+cpu
           pip list
       - name: Run ${{ matrix.name }} Tests
         run: ./run_tests.sh ${{ matrix.framework }}