diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 895f7362b..1fdc37efa 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -42,14 +42,22 @@ jobs: policy: "manylinux2014" platform: "x86_64" CUDA_BASE_IMAGE: "nvidia/cuda:11.3.1-cudnn8-devel-centos7" + target: "manylinux-with-trt" - tag-suffix: "cuda11.2" policy: "manylinux2014" platform: "x86_64" CUDA_BASE_IMAGE: "nvidia/cuda:11.2.2-cudnn8-devel-centos7" + target: "manylinux" + - tag-suffix: "cuda11.0" + policy: "manylinux2014" + platform: "x86_64" + CUDA_BASE_IMAGE: "nvidia/cuda:11.0.3-cudnn8-devel-centos7" + target: "manylinux" - tag-suffix: "cuda10.2" policy: "manylinux2014" platform: "x86_64" CUDA_BASE_IMAGE: "nvidia/cuda:10.2-cudnn7-devel-centos7" + target: "manylinux" - tag-suffix: "cpu" policy: "manylinux2014" platform: "x86_64" @@ -102,6 +110,7 @@ jobs: cache-from: type=registry,ref=${{ env.DOCKER_HUB_NAMESPACE }}/${{ env.DOCKER_REPO }}:latest cache-to: type=inline context: ./docker/ + target: ${ matrix.target } build-args: | POLICY PLATFORM diff --git a/docker/Dockerfile b/docker/Dockerfile index e2f0bc3ee..a7e1545d5 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -149,7 +149,7 @@ COPY --from=build_cpython310 /opt/_internal /opt/_internal/ RUN manylinux-entrypoint /build_scripts/finalize-python.sh -FROM runtime_base +FROM runtime_base AS manylinux COPY --from=build_git /manylinux-rootfs / COPY --from=build_swig /manylinux-rootfs / COPY --from=build_cpython /manylinux-rootfs / @@ -174,4 +174,8 @@ RUN curl -L $BAZEL_URL -o /usr/local/bin/bazel \ && chmod +x /usr/local/bin/bazel RUN yum install -y wget nasm rdma-core-devel rsync gdb ninja-build openblas-static devtoolset-7-gcc* vim ccache htop +FROM manylinux AS manylinux-with-trt +ARG trt=... +RUN install trt ${} + CMD ["/bin/bash"]