support torch 2.2.0

This commit is contained in:
Fangjun Kuang 2024-02-10 17:12:40 +08:00
parent 936767d8c1
commit ed9e645195
3 changed files with 13 additions and 7 deletions

View File

@ -3,7 +3,7 @@ name: build-wheels-cpu-win64
on: on:
push: push:
branches: branches:
- fix-wheel - fix-wheel-2
tags: tags:
- '*' - '*'
workflow_dispatch: workflow_dispatch:

View File

@ -26,10 +26,10 @@ 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}.3" echo "Installing ${PYTHON_VERSION}.2"
curl -O https://www.python.org/ftp/python/${PYTHON_VERSION}.3/Python-${PYTHON_VERSION}.3.tgz curl -O https://www.python.org/ftp/python/${PYTHON_VERSION}.2/Python-${PYTHON_VERSION}.2.tgz
tar xf Python-${PYTHON_VERSION}.3.tgz tar xf Python-${PYTHON_VERSION}.2.tgz
pushd Python-${PYTHON_VERSION}.3 pushd Python-${PYTHON_VERSION}.2
PYTHON_INSTALL_DIR=$PWD/py-${PYTHON_VERSION} PYTHON_INSTALL_DIR=$PWD/py-${PYTHON_VERSION}
@ -44,7 +44,7 @@ make install >/dev/null 2>&1
popd popd
echo "pwd: $PWD" echo "pwd: $PWD"
# rm -rf Python-${PYTHON_VERSION}.3 # rm -rf Python-${PYTHON_VERSION}.2
export PATH=$PYTHON_INSTALL_DIR/bin:$PATH export PATH=$PYTHON_INSTALL_DIR/bin:$PATH
export LD_LIBRARY_PATH=$PYTHON_INSTALL_DIR/lib:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=$PYTHON_INSTALL_DIR/lib:$LD_LIBRARY_PATH

View File

@ -165,10 +165,16 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"],
}, },
"2.2.0": {
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
"cuda": ["11.8", "12.1"] # default 12.1
if not for_windows
else ["11.8.0", "12.1.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.1.2" latest = "2.2.0"
matrix = {latest: matrix[latest]} matrix = {latest: matrix[latest]}
if for_windows or for_macos: if for_windows or for_macos: