diff --git a/.github/workflows/build_dynamic_embedding_wheels.yml b/.github/workflows/build_dynamic_embedding_wheels.yml index 339e8e734..3abc051bc 100644 --- a/.github/workflows/build_dynamic_embedding_wheels.yml +++ b/.github/workflows/build_dynamic_embedding_wheels.yml @@ -20,8 +20,8 @@ jobs: fail-fast: false matrix: os: [ ubuntu-latest ] - pyver: [ cp39, cp310 ] - cuver: [ "11.8" ] + pyver: [ cp39, cp310, cp311, cp312 ] + cuver: [ "12.1" ] steps: - diff --git a/contrib/dynamic_embedding/tools/before_linux_build.sh b/contrib/dynamic_embedding/tools/before_linux_build.sh index 9badc61d7..14cba7d0c 100755 --- a/contrib/dynamic_embedding/tools/before_linux_build.sh +++ b/contrib/dynamic_embedding/tools/before_linux_build.sh @@ -9,7 +9,7 @@ set -xe distro=rhel7 arch=x86_64 -CUDA_VERSION="${CUDA_VERSION:-11.8}" +CUDA_VERSION="${CUDA_VERSION:-12.1}" CUDA_MAJOR_VERSION=$(echo "${CUDA_VERSION}" | tr '.' ' ' | awk '{print $1}') CUDA_MINOR_VERSION=$(echo "${CUDA_VERSION}" | tr '.' ' ' | awk '{print $2}')