Skip to content

Commit f64adc0

Browse files
committed
Add gcc 10 to the list of allowed compilers
1 parent 5a21cd9 commit f64adc0

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

tensorflow_cc/cmake/build_tensorflow.sh.in

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,11 @@ 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-9 2>/dev/null && version_gt 9.4 `gcc-9 -dumpversion`; then
66+
if hash gcc 2>/dev/null && version_gt 10.3 `gcc -dumpversion`; then
67+
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc"}
68+
elif hash gcc-10 2>/dev/null && version_gt 10.3 `gcc-10 -dumpversion`; then
69+
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc-10"}
70+
elif hash gcc-9 2>/dev/null && version_gt 9.4 `gcc-9 -dumpversion`; then
6771
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc-9"}
6872
elif hash gcc-8 2>/dev/null && version_gt 8.5 `gcc-8 -dumpversion`; then
6973
export GCC_HOST_COMPILER_PATH=${GCC_HOST_COMPILER_PATH:-"/usr/bin/gcc-8"}

0 commit comments

Comments
 (0)