Enable support for Ubuntu 24.04 (#107) (#108)

* Update python version * Update path for RHEL
triton-inference-server · Nov 26, 2024 · 0a66a4c · 0a66a4c
1 parent e4fe6b8
commit 0a66a4c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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'")