From e751d10412da1ac245614c40ce59e81c0195e6a8 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Mon, 4 Nov 2024 21:22:07 -0800 Subject: [PATCH] Update path for RHEL --- CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index eb1188d..d0682a6 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -77,7 +77,7 @@ 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.12.3/lib/python3.12/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.12/dist-packages/tensorflow") endif(RHEL_BUILD)