From cab13fa67ccfd908b6b1331863b67164ef65259b Mon Sep 17 00:00:00 2001 From: Henry Tsang Date: Mon, 11 Dec 2023 21:42:27 -0800 Subject: [PATCH] update dynamic embedding github actions from 116 to 118 (#1576) Summary: Pull Request resolved: https://github.com/pytorch/torchrec/pull/1576 Differential Revision: D52063343 --- .github/workflows/build_dynamic_embedding_wheels.yml | 2 +- contrib/dynamic_embedding/tools/before_linux_build.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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}')