Specify torchvision version in run_pytorch_tests (#108)

Specify torchvision version in run_pytorch_tests and pass the needed version in the workflows that use it.

.github/workflows/run_pytorch_tests.yml

@@ -6,6 +6,10 @@ on:
       torch-version:
         required: true
         type: string
+      torchvision-version:
+        required: true
+        type: string
+
 
 jobs:
   run-pytorch-tests:
@@ -20,7 +24,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt          
-          pip install torch==${{ inputs.torch-version }} torchvision onnx onnxruntime onnxruntime-extensions
+          pip install torch==${{ inputs.torch-version }} torchvision==${{ inputs.torchvision-version }} onnx onnxruntime onnxruntime-extensions
       - name: Run unittests
         run: python -m unittest discover tests/pytorch_tests -v
 

.github/workflows/run_tests_torch2_0.yml

@@ -9,3 +9,4 @@ jobs:
     uses: ./.github/workflows/run_pytorch_tests.yml
     with:
       torch-version: "2.0.*"
+      torchvision-version: "0.15.*"

.github/workflows/run_tests_torch2_1.yml

@@ -9,3 +9,4 @@ jobs:
     uses: ./.github/workflows/run_pytorch_tests.yml
     with:
       torch-version: "2.1.*"
+      torchvision-version: "0.16.*"

.github/workflows/run_tests_torch2_2.yml

@@ -9,3 +9,4 @@ jobs:
     uses: ./.github/workflows/run_pytorch_tests.yml
     with:
       torch-version: "2.2.*"
+      torchvision-version: "0.17.*"

.github/workflows/run_tests_torch2_3.yml

@@ -9,3 +9,4 @@ jobs:
     uses: ./.github/workflows/run_pytorch_tests.yml
     with:
       torch-version: "2.3.*"
+      torchvision-version: "0.18.*"

.github/workflows/run_tests_torch2_4.yml

@@ -9,3 +9,4 @@ jobs:
     uses: ./.github/workflows/run_pytorch_tests.yml
     with:
       torch-version: "2.4.*"
+      torchvision-version: "0.19.*"