@@ -208,20 +208,20 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
208208 COMMAND docker rm pytorch_backend_ptlib || echo "error ignored..." || true
209209 COMMAND docker create --name pytorch_backend_ptlib ${TRITON_PYTORCH_DOCKER_IMAGE}
210210 COMMAND /bin/sh -c "for i in ${LIBTORCH_LIBS_STR} ; do echo copying $i && docker cp -L pytorch_backend_ptlib:/usr/local/lib/$i $i ; done"
211- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/lib/libc10.so libc10.so
212- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/lib/libc10_cuda.so libc10_cuda.so
213- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/lib/libtorch.so libtorch.so
214- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/lib/libtorch_cpu.so libtorch_cpu.so
215- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/lib/libtorch_cuda.so libtorch_cuda.so
216- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/lib/libtorch_cuda_linalg.so libtorch_cuda_linalg.so
217- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/lib/libnvfuser_codegen.so libnvfuser_codegen.so
218- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/lib/libtorch_global_deps.so libtorch_global_deps.so
219- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/lib/libcaffe2_nvrtc.so libcaffe2_nvrtc.so
211+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/lib/libc10.so libc10.so
212+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/lib/libc10_cuda.so libc10_cuda.so
213+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/lib/libtorch.so libtorch.so
214+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/lib/libtorch_cpu.so libtorch_cpu.so
215+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/lib/libtorch_cuda.so libtorch_cuda.so
216+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/lib/libtorch_cuda_linalg.so libtorch_cuda_linalg.so
217+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/lib/libnvfuser_codegen.so libnvfuser_codegen.so
218+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/lib/libtorch_global_deps.so libtorch_global_deps.so
219+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/lib/libcaffe2_nvrtc.so libcaffe2_nvrtc.so
220220 COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/libtorchvision.so libtorchvision.so
221- COMMAND /bin/sh -c "if [ ${TRITON_PYTORCH_ENABLE_TORCHTRT} = 'ON' ]; then docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch_tensorrt/lib/libtorchtrt_runtime.so libtorchtrt_runtime.so; fi"
222- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch_tensorrt/bin/torchtrtc torchtrtc || echo "error ignored..." || true
221+ 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"
222+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch_tensorrt/bin/torchtrtc torchtrtc || echo "error ignored..." || true
223223 COMMAND docker cp pytorch_backend_ptlib:/opt/pytorch/pytorch/LICENSE LICENSE.pytorch
224- COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.8 /dist-packages/torch/include include /torch
224+ COMMAND docker cp pytorch_backend_ptlib:/usr/local/lib/python3.10 /dist-packages/torch/include include /torch
225225 COMMAND docker cp pytorch_backend_ptlib:/opt/pytorch/pytorch/torch/csrc/jit/codegen include /torch/torch/csrc/jit/.
226226 COMMAND docker cp pytorch_backend_ptlib:/opt/pytorch/vision/torchvision/csrc include /torchvision/torchvision
227227 COMMAND docker cp -L pytorch_backend_ptlib:/usr/local/lib/libopencv_videoio.so libopencv_videoio.so
0 commit comments