mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-09 18:12:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

support torch 2.5.0

Browse Source
This commit is contained in:
Fangjun Kuang 2024-10-18 10:28:57 +08:00
parent e05540cfad
commit ca22057a86
4 changed files with 47 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/github_actions/build-ubuntu-cpu.sh
View File

@ -17,11 +17,15 @@ fi
yum -y install openssl-devel bzip2-devel libffi-devel xz-devel wget redhat-lsb-core yum -y install openssl-devel bzip2-devel libffi-devel xz-devel wget redhat-lsb-core
echo "Installing ${PYTHON_VERSION}.2" INSTALLED_PYTHON_VERSION=${PYTHON_VERSION}.2
if [[ $PYTHON_VERSION == "3.13" ]]; then
INSTALLED_PYTHON_VERSION=${PYTHON_VERSION}.0
fi
echo "Installing $INSTALLED_PYTHON_VERSION"
curl -O https://www.python.org/ftp/python/${PYTHON_VERSION}.2/Python-${PYTHON_VERSION}.2.tgz curl -O https://www.python.org/ftp/python/$INSTALLED_PYTHON_VERSION/Python-$INSTALLED_PYTHON_VERSION.tgz
tar xf Python-${PYTHON_VERSION}.2.tgz tar xf Python-$INSTALLED_PYTHON_VERSION.tgz
pushd Python-${PYTHON_VERSION}.2 pushd Python-$INSTALLED_PYTHON_VERSION
PYTHON_INSTALL_DIR=$PWD/py-${PYTHON_VERSION} PYTHON_INSTALL_DIR=$PWD/py-${PYTHON_VERSION}

13
scripts/github_actions/build-ubuntu-cuda.sh
View File

@ -31,10 +31,15 @@ fi
yum -y install openssl-devel bzip2-devel libffi-devel xz-devel wget redhat-lsb-core yum -y install openssl-devel bzip2-devel libffi-devel xz-devel wget redhat-lsb-core
echo "Installing ${PYTHON_VERSION}.2" INSTALLED_PYTHON_VERSION=${PYTHON_VERSION}.2
curl -O https://www.python.org/ftp/python/${PYTHON_VERSION}.2/Python-${PYTHON_VERSION}.2.tgz if [[ $PYTHON_VERSION == "3.13" ]]; then
tar xf Python-${PYTHON_VERSION}.2.tgz INSTALLED_PYTHON_VERSION=${PYTHON_VERSION}.0
pushd Python-${PYTHON_VERSION}.2 fi
echo "Installing $INSTALLED_PYTHON_VERSION"
curl -O https://www.python.org/ftp/python/$INSTALLED_PYTHON_VERSION/Python-$INSTALLED_PYTHON_VERSION.tgz
tar xf Python-$INSTALLED_PYTHON_VERSION.tgz
pushd Python-$INSTALLED_PYTHON_VERSION
PYTHON_INSTALL_DIR=$PWD/py-${PYTHON_VERSION} PYTHON_INSTALL_DIR=$PWD/py-${PYTHON_VERSION}

14
scripts/github_actions/generate_build_matrix.py
View File

@ -259,13 +259,25 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
else ["11.8.0", "12.1.0", "12.4.0"] else ["11.8.0", "12.1.0", "12.4.0"]
), ),
}, },
"2.5.0": {
# Only Linux support python 3.13
"python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"],
"cuda": (
["11.8", "12.1", "12.4"] # default 12.4
if not for_windows
else ["11.8.0", "12.1.0", "12.4.0"]
),
},
# https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts # https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
} }
if test_only_latest_torch: if test_only_latest_torch:
latest = "2.4.1" latest = "2.5.0"
matrix = {latest: matrix[latest]} matrix = {latest: matrix[latest]}
if for_windows or for_macos: if for_windows or for_macos:
if "2.5.0" in matrix:
matrix["2.5.0"]["python-version"].remove("3.13")
if "1.13.0" in matrix: if "1.13.0" in matrix:
matrix["1.13.0"]["python-version"].remove("3.11") matrix["1.13.0"]["python-version"].remove("3.11")

17
scripts/github_actions/install_torch.sh
View File

@ -230,6 +230,23 @@ case ${torch} in
;; ;;
esac esac
;; ;;
2.5.*)
case ${cuda} in
11.8)
package="torch==${torch}+cu118"
# https://download.pytorch.org/whl/nightly/torch/
url=https://download.pytorch.org/whl/torch/
;;
12.1)
package="torch==${torch}+cu121"
url=https://download.pytorch.org/whl/torch/
;;
12.4)
package="torch==${torch}+cu124"
url=https://download.pytorch.org/whl/torch/
;;
esac
;;
*) *)
echo "Unsupported PyTorch version: ${torch}" echo "Unsupported PyTorch version: ${torch}"
exit 1 exit 1