@@ -89,15 +89,15 @@ find_package(Python3 REQUIRED COMPONENTS Development.Module)
8989set (RHEL_BUILD OFF )
9090set (LIB_DIR "lib" )
9191set (LIBTORCH_LIBS_PATH "/usr/local/lib" )
92- set (PY_INSTALL_PATH "/usr/local/lib/python3.10 /dist-packages" )
92+ set (PY_INSTALL_PATH "/usr/local/lib/python3.12 /dist-packages" )
9393if (LINUX)
9494 file (STRINGS "/etc/os-release" DISTRO_ID_LIKE REGEX "ID_LIKE" )
9595 if (${DISTRO_ID_LIKE} MATCHES "rhel|centos" )
9696 set (RHEL_BUILD ON )
9797 set (LIB_DIR "lib64" )
98- set (PY_INSTALL_PATH "/opt/_internal/cpython-3.10.13 /lib/python3.10 /site-packages" )
98+ set (PY_INSTALL_PATH "/opt/_internal/cpython-3.12.3 /lib/python3.12 /site-packages" )
9999 if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "x86_64" )
100- set (LIBTORCH_LIBS_PATH "/opt/_internal/cpython-3.10.13 /lib" )
100+ set (LIBTORCH_LIBS_PATH "/opt/_internal/cpython-3.12.3 /lib" )
101101 endif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "x86_64" )
102102 endif (${DISTRO_ID_LIKE} MATCHES "rhel|centos" )
103103endif (LINUX)
@@ -249,7 +249,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
249249 COMMAND docker cp -L pytorch_backend_ptlib:/usr/local/cuda/lib64/libcusparseLt.so libcusparseLt.so;
250250 COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_ENABLE_TORCHVISION} = 'ON' ]; then docker cp pytorch_backend_ptlib:/usr/local/${LIB_DIR} /libtorchvision.so libtorchvision.so; fi"
251251 COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_ENABLE_TORCHVISION} = 'ON' ]; then docker cp pytorch_backend_ptlib:/opt/pytorch/vision/torchvision/csrc include/torchvision/torchvision; fi"
252- COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_ENABLE_TORCHTRT} = 'ON' ]; then docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch_tensorrt/lib/libtorchtrt_runtime.so libtorchtrt_runtime.so; fi"
252+ COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_ENABLE_TORCHTRT} = 'ON' ]; then docker cp pytorch_backend_ptlib:/usr/local/lib/python3.12 /dist-packages/torch_tensorrt/lib/libtorchtrt_runtime.so libtorchtrt_runtime.so; fi"
253253 COMMAND docker cp pytorch_backend_ptlib:${PY_INSTALL_PATH} /torch_tensorrt/bin/torchtrtc torchtrtc || echo "error ignored..." || true
254254 COMMAND docker cp pytorch_backend_ptlib:/opt/pytorch/pytorch/LICENSE LICENSE.pytorch
255255 COMMAND docker cp pytorch_backend_ptlib:${PY_INSTALL_PATH} /torch/include include /torch
0 commit comments