Add tests for PyTorch 1.10 - 2.1 (#1218)

Summary:
Adds tests for missing versions of PyTorch to make sure tests cover all supported PyTorch versions.

Pull Request resolved: #1218

Reviewed By: aobo-y

Differential Revision: D51823341

Pulled By: vivekmig

fbshipit-source-id: 395836ca7683046c99ec2aeaf90c3dd65b1da37b

Author: vivekmig

.github/workflows/test-pip-cpu.yml

@@ -12,12 +12,23 @@ jobs:
   tests:
     strategy:
       matrix:
-        pytorch_args: ["-v 1.6", "-v 1.7", " -v 1.8", "-v 1.9"]
+        pytorch_args: ["-v 1.6", "-v 1.7", "-v 1.8", "-v 1.9", "-v 1.10", "-v 1.11", "-v 1.12", "-v 1.13"]
+        docker_img: ["cimg/python:3.6", "cimg/python:3.7"]
+        include:
+          - pytorch_args: "-v 2.0"
+            docker_img: "cimg/python:3.8"
+        exclude:
+          - pytorch_args: "-v 1.11"
+            docker_img: "cimg/python:3.6"
+          - pytorch_args: "-v 1.12"
+            docker_img: "cimg/python:3.6"
+          - pytorch_args: "-v 1.13"
+            docker_img: "cimg/python:3.6"
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: linux.12xlarge
-      docker-image: cimg/python:3.6
+      docker-image: ${{ matrix.docker_img }}
       repository: pytorch/captum
       script: |
         sudo chmod -R 777 .