Update cuda toolchain for manylinux2014

tensorflow · Apr 19, 2022 · 55700f9 · 55700f9
1 parent 54d69d0
commit 55700f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/install_so_files.sh b/tools/install_so_files.sh
@@ -1,7 +1,7 @@
 set -e -x
 
 if [ "$TF_NEED_CUDA" == "1" ]; then
-  CUDA_FLAG="--crosstool_top=//build_deps/toolchains/gcc7_manylinux2010-nvcc-cuda11:toolchain"
+  CUDA_FLAG="--crosstool_top=@ubuntu20.04-gcc9_manylinux2014-cuda11.2-cudnn8.1-tensorrt7.2_config_cuda//crosstool:toolchain"
 fi
 
 bazel build $CUDA_FLAG //tensorflow_addons/...