Put back special case for python 3.11 - fix tensorrt

pytorch · Oct 12, 2023 · c45f241 · c45f241
1 parent cdae9de
commit c45f241
Showing 1 changed file with 32 additions and 9 deletions.
diff --git a/.github/actions/setup-binary-builds/action.yml b/.github/actions/setup-binary-builds/action.yml
@@ -109,16 +109,39 @@ runs:
           set -euxo pipefail
           CONDA_ENV="${RUNNER_TEMP}/conda_environment_${GITHUB_RUN_ID}"
 
-          conda create \
-            --yes --quiet \
-            --prefix "${CONDA_ENV}" \
-            "python=${PYTHON_VERSION}" \
-            cmake=3.22 \
-            conda-build=3.21 \
-            ninja=1.10 \
-            pkg-config=0.29 \
-            wheel=0.37
+          if [[ "${PYTHON_VERSION}" = "3.11" ]]; then
 
+            if [[ "${PACKAGE_TYPE:-}" == "conda" ]]; then
+              conda create \
+                --yes --quiet \
+                --prefix "${CONDA_ENV}" \
+                "python=3.8" \
+                cmake=3.22 \
+                conda-build=3.21 \
+                ninja=1.10 \
+                pkg-config=0.29 \
+                wheel=0.37
+            else
+              conda create \
+                --yes --quiet \
+                --prefix "${CONDA_ENV}" \
+                "python=${PYTHON_VERSION}" \
+                cmake=3.22 \
+                ninja=1.10 \
+                pkg-config=0.29 \
+                wheel=0.37
+            fi
+          else
+            conda create \
+              --yes --quiet \
+              --prefix "${CONDA_ENV}" \
+              "python=${PYTHON_VERSION}" \
+              cmake=3.22 \
+              conda-build=3.21 \
+              ninja=1.10 \
+              pkg-config=0.29 \
+              wheel=0.37
+          fi
 
           echo "CONDA_ENV=${CONDA_ENV}" >> "${GITHUB_ENV}"
           echo "CONDA_RUN=conda run -p ${CONDA_ENV}" >> "${GITHUB_ENV}"