diff --git a/.github/workflows/build_dynamic_embedding_wheels.yml b/.github/workflows/build_dynamic_embedding_wheels.yml index bd4fd121f..8dd7231b0 100644 --- a/.github/workflows/build_dynamic_embedding_wheels.yml +++ b/.github/workflows/build_dynamic_embedding_wheels.yml @@ -21,7 +21,7 @@ jobs: matrix: os: [ ubuntu-latest ] pyver: [ cp38, cp39, cp310 ] - cuver: [ "11.6" ] + cuver: [ "11.8" ] steps: - diff --git a/contrib/dynamic_embedding/tools/before_linux_build.sh b/contrib/dynamic_embedding/tools/before_linux_build.sh index 7cf1d154f..3b94ddcaf 100755 --- a/contrib/dynamic_embedding/tools/before_linux_build.sh +++ b/contrib/dynamic_embedding/tools/before_linux_build.sh @@ -3,7 +3,7 @@ set -xe distro=rhel7 arch=x86_64 -CUDA_VERSION="${CUDA_VERSION:-11.6}" +CUDA_VERSION="${CUDA_VERSION:-11.8}" CUDA_MAJOR_VERSION=$(echo "${CUDA_VERSION}" | tr '.' ' ' | awk '{print $1}') CUDA_MINOR_VERSION=$(echo "${CUDA_VERSION}" | tr '.' ' ' | awk '{print $2}')