@@ -208,20 +208,20 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
208
208
COMMAND docker rm pytorch_backend_ptlib || echo "error ignored..." || true
209
209
COMMAND docker create --name pytorch_backend_ptlib ${TRITON_PYTORCH_DOCKER_IMAGE}
210
210
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
220
220
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
223
223
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
225
225
COMMAND docker cp pytorch_backend_ptlib:/opt/pytorch/pytorch/torch/csrc/jit/codegen include /torch/torch/csrc/jit/.
226
226
COMMAND docker cp pytorch_backend_ptlib:/opt/pytorch/vision/torchvision/csrc include /torchvision/torchvision
227
227
COMMAND docker cp -L pytorch_backend_ptlib:/usr/local/lib/libopencv_videoio.so libopencv_videoio.so
0 commit comments