Skip to content

Commit 1e695f1

Browse files
committed
Allow gcc-9
1 parent 56d94c7 commit 1e695f1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tensorflow_cc/cmake/build_tensorflow.sh.in

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,9 @@ if [ "$cuda_allowed" == true ] && [ "$cuda_available" == true ]; then
6363

6464
# choose the right version of CUDA compiler
6565
if [ -z "$GCC_HOST_COMPILER_PATH" ]; then
66-
if hash gcc-8 2>/dev/null && version_gt 8.5 `gcc-8 -dumpversion`; then
66+
if hash gcc-9 2>/dev/null && version_gt 9.4 `gcc-9 -dumpversion`; then
67+
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc-9"}
68+
elif hash gcc-8 2>/dev/null && version_gt 8.5 `gcc-8 -dumpversion`; then
6769
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc-8"}
6870
elif hash gcc-7 2>/dev/null && version_gt 7.5 `gcc-7 -dumpversion`; then
6971
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc-7"}

0 commit comments

Comments
 (0)