Unpin Python binaries from Python 3.8 (#2108)

Summary: Pull Request resolved: #2108 Unpin Python binaries from Python 3.8. This is a re-attempt of D50814040 Reviewed By: spcyppt Differential Revision: D50907864 fbshipit-source-id: d190b43e38b46bd9173077d75e1b8b7b2f58aa90
pytorch · Nov 2, 2023 · cbb3130 · cbb3130
1 parent 824ef10
commit cbb3130
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.github/scripts/fbgemm_gpu_test.bash b/.github/scripts/fbgemm_gpu_test.bash
@@ -138,7 +138,7 @@ test_setup_conda_environment () {
   if [ "$pytorch_variant_type" == "" ]; then
     echo "Usage: ${FUNCNAME[0]} ENV_NAME PYTHON_VERSION PYTORCH_INSTALLER PYTORCH_VERSION PYTORCH_VARIANT_TYPE [PYTORCH_VARIANT_VERSION]"
     echo "Example(s):"
-    echo "    ${FUNCNAME[0]} build_env 3.8 pip test cuda 11.8.0       # Setup environment with pytorch-test for Python 3.8 + CUDA 11.8.0"
+    echo "    ${FUNCNAME[0]} build_env 3.10 pip test cuda 12.1.0       # Setup environment with pytorch-test for Python 3.10 + CUDA 12.1.0"
     return 1
   else
     echo "################################################################################"

diff --git a/.github/scripts/test_torchrec.bash b/.github/scripts/test_torchrec.bash
@@ -27,9 +27,9 @@ usage () {
   echo "PACKAGE_NAME        : output package name of TorchRec (e.g., torchrec_nightly)"
   echo "                      Note: TorchRec is sensitive to its package name"
   echo "                      e.g., torchrec needs fbgemm-gpu while torchrec_nightly needs fbgemm-gpu-nightly"
-  echo "PYTHON_VERSION      : Python version (e.g., 3.8, 3.9, 3.10)"
+  echo "PYTHON_VERSION      : Python version (e.g., 3.10)"
   echo "PYTORCH_CHANNEL_NAME: PyTorch's channel name (e.g., pytorch-nightly, pytorch-test (=pre-release), pytorch (=stable release))"
-  echo "CUDA_VERSION        : PyTorch's CUDA version (e.g., 11.6, 11.7)"
+  echo "CUDA_VERSION        : PyTorch's CUDA version (e.g., 12.1)"
   echo "FBGEMM_WHEEL_PATH   : path to FBGEMM_GPU's wheel file"
   echo "MINICONDA_PREFIX    : path to install Miniconda (default: \$HOME/miniconda)"
   echo "Example: Python 3.10 + PyTorch nightly (CUDA 12.1), install miniconda at \$HOME/miniconda, using dist/fbgemm_gpu_nightly.whl"

diff --git a/fbgemm_gpu/setup.py b/fbgemm_gpu/setup.py
@@ -387,6 +387,7 @@ def main(argv: List[str]) -> None:
             "Programming Language :: Python :: 3.8",
             "Programming Language :: Python :: 3.9",
             "Programming Language :: Python :: 3.10",
+            "Programming Language :: Python :: 3.11",
             "Topic :: Scientific/Engineering :: Artificial Intelligence",
         ],
     )