Skip to content

Commit

Permalink
Update path for RHEL
Browse files Browse the repository at this point in the history
  • Loading branch information
mc-nv committed Nov 5, 2024
1 parent d975042 commit e751d10
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit e751d10

Please sign in to comment.