update dynamic embedding github actions from 116 to 118 (pytorch#1576)

Summary: Pull Request resolved: pytorch#1576 Differential Revision: D52063343
henrylhtsang · Dec 12, 2023 · cab13fa · cab13fa
1 parent befa94b
commit cab13fa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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}')