Skip to content

Commit 4d41c1b

Browse files
committed
Update Python version
1 parent e97a900 commit 4d41c1b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

CMakeLists.txt

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -89,15 +89,15 @@ find_package(Python3 REQUIRED COMPONENTS Development.Module)
8989
set(RHEL_BUILD OFF)
9090
set(LIB_DIR "lib")
9191
set(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")
9393
if(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")
103103
endif(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

Comments
 (0)