Add 3.13t wheel builds (pytorch#8898)

NicolasHug · Feb 5, 2025 · 927c4e3 · 927c4e3
1 parent 947722a
commit 927c4e3
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/packaging/pre_build_script.sh b/packaging/pre_build_script.sh
@@ -1,5 +1,12 @@
 #!/bin/bash
 
+CONDA_CHANNEL=""
+
+# python python-freethreading build use conda-forge
+if [[ "${PYTHON_VERSION: -1}" == "t" ]]; then
+  CONDA_CHANNEL="-c conda-forge"
+fi
+
 if [[ "$(uname)" == Darwin ]]; then
   # Uninstall Conflicting jpeg brew formulae
   jpeg_packages=$(brew list | grep jpeg)
@@ -13,7 +20,7 @@ if [[ "$(uname)" == Darwin ]]; then
 fi
 
 if [[ "$(uname)" == Darwin || "$OSTYPE" == "msys" ]]; then
-  conda install libpng libwebp -yq
+  conda install -yv libpng libwebp ${CONDA_CHANNEL}
   # Installing webp also installs a non-turbo jpeg, so we uninstall jpeg stuff
   # before re-installing them
   conda uninstall libjpeg-turbo libjpeg -y
@@ -28,11 +35,11 @@ if [[ "$(uname)" == Darwin || "$OSTYPE" == "msys" ]]; then
 else
 
   if [[ "$ARCH" == "aarch64" ]]; then
-    conda install libpng -yq
+    conda install -yv libpng ${CONDA_CHANNEL}
     conda install -yq ffmpeg=4.2 libjpeg-turbo -c pytorch-nightly
   fi
 
-  conda install libwebp -yq
+  conda install -v libwebp ${CONDA_CHANNEL}
   conda install libjpeg-turbo -c pytorch
   yum install -y freetype gnutls
   pip install auditwheel