Skip to content

Commit 21e055f

Browse files
committed
Make sure only allowed gcc versions are used
1 parent 83d079d commit 21e055f

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tensorflow_cc/cmake/build_tensorflow.sh.in

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -61,9 +61,7 @@ if [ "$cuda_allowed" == true ] && [ "$cuda_available" == true ]; then
6161

6262
# choose the right version of CUDA compiler
6363
if [ -z "$GCC_HOST_COMPILER_PATH" ]; then
64-
if hash gcc 2>/dev/null && version_gt 10.3 `gcc -dumpversion`; then
65-
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc"}
66-
elif hash gcc-10 2>/dev/null && version_gt 10.3 `gcc-10 -dumpversion`; then
64+
if hash gcc-10 2>/dev/null && version_gt 10.3 `gcc-10 -dumpversion`; then
6765
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc-10"}
6866
elif hash gcc-9 2>/dev/null && version_gt 9.4 `gcc-9 -dumpversion`; then
6967
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc-9"}

0 commit comments

Comments
 (0)