Bump PyTorch and Python version (#1298)

Summary: Pull Request resolved: #1298

Differential Revision: D58695682

Author: yucu

.conda/meta.yaml

@@ -13,10 +13,10 @@ build:
 
 requirements:
   host:
-    - python>=3.6
+    - python>=3.8
   run:
-    - numpy
-    - pytorch>=1.6
+    - numpy<2.0
+    - pytorch>=1.13
     - matplotlib-base
 
 test:

.github/workflows/lint.yml

@@ -13,7 +13,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: linux.12xlarge
-      docker-image: cimg/python:3.9
+      docker-image: cimg/python:3.11
       repository: pytorch/captum
       script: |
         sudo chmod -R 777 .

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

@@ -15,7 +15,7 @@ jobs:
   tests:
     strategy:
       matrix:
-        python_version: ["3.7", "3.8", "3.9", "3.10"]
+        python_version: ["3.8", "3.9", "3.10", "3.11"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:

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

@@ -17,7 +17,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: linux.12xlarge
-      docker-image: cimg/python:3.6
+      docker-image: cimg/python:3.11
       repository: pytorch/captum
       script: |
         sudo chmod -R 777 .

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

@@ -12,18 +12,8 @@ jobs:
   tests:
     strategy:
       matrix:
-        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"
+        pytorch_args: ["-v 1.13", "-v 2.0.0", "-v 2.1.0", "-v 2.2.0", "-v 2.3.0"]
+        docker_img: ["cimg/python:3.8", "cimg/python:3.9", "cimg/python:3.10", "cimg/python:3.11"]
       fail-fast: false
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:

README.md

@@ -47,8 +47,8 @@ Captum can also be used by application engineers who are using trained models in
 ## Installation
 
 **Installation Requirements**
-- Python >= 3.6
-- PyTorch >= 1.6
+- Python >= 3.8
+- PyTorch >= 1.13
 
 
 ##### Installing the latest release

environment.yml

@@ -2,5 +2,5 @@ name: captum
 channels:
   - pytorch
 dependencies:
-  - numpy
-  - pytorch>=1.6
+  - numpy<2.0
+  - pytorch>=1.13

setup.py

@@ -146,8 +146,8 @@ def get_package_files(root, subdirs):
         ],
         long_description=long_description,
         long_description_content_type="text/markdown",
-        python_requires=">=3.6",
-        install_requires=["matplotlib", "numpy", "torch>=1.6", "tqdm"],
+        python_requires=">=3.8",
+        install_requires=["matplotlib", "numpy<2.0", "torch>=1.13", "tqdm"],
         packages=find_packages(exclude=("tests", "tests.*")),
         extras_require={
             "dev": DEV_REQUIRES,