diff --git a/CMakeLists.txt b/CMakeLists.txt index 24ebed5..d0682a6 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -77,9 +77,9 @@ elseif(${TRITON_TENSORFLOW_VERSION} EQUAL "2") set(TRITON_TENSORFLOW_BACKEND_LIBNAME triton_tensorflow) set(TRITON_TENSORFLOW_BACKEND_INSTALLDIR ${CMAKE_INSTALL_PREFIX}/backends/tensorflow) if(RHEL_BUILD) - set(TRITON_TENSORFLOW_PYTHON_PATH "/opt/_internal/cpython-3.10.13/lib/python3.10/site-packages/tensorflow") + set(TRITON_TENSORFLOW_PYTHON_PATH "/opt/_internal/cpython-3.12.1/lib/python3.12/site-packages/tensorflow") else() - set(TRITON_TENSORFLOW_PYTHON_PATH "/usr/local/lib/python3.10/dist-packages/tensorflow") + set(TRITON_TENSORFLOW_PYTHON_PATH "/usr/local/lib/python3.12/dist-packages/tensorflow") endif(RHEL_BUILD) else() message(FATAL_ERROR "TRITON_TENSORFLOW_VERSION allowed values is '2'")