diff --git a/.github/workflows/_build_linux.yml b/.github/workflows/_build_linux.yml index 9f015fae2e0..65f12e0b314 100644 --- a/.github/workflows/_build_linux.yml +++ b/.github/workflows/_build_linux.yml @@ -55,7 +55,7 @@ on: jobs: fd-build: runs-on: [self-hosted, GPU-Build] - timeout-minutes: 240 + timeout-minutes: 360 outputs: wheel_path: ${{ steps.set_output.outputs.wheel_path }} steps: diff --git a/.github/workflows/_ci_image_build.yml b/.github/workflows/_ci_image_build.yml index 92ff8b6077c..99b21dad18c 100644 --- a/.github/workflows/_ci_image_build.yml +++ b/.github/workflows/_ci_image_build.yml @@ -29,7 +29,7 @@ jobs: outputs: docker_name_precheck: ${{ steps.docker_build.outputs.docker_name_precheck }} steps: - - name: Code Prepare + - name: Docker Build id: docker_build shell: bash env: diff --git a/.github/workflows/_unit_test_coverage.yml b/.github/workflows/_unit_test_coverage.yml index f5c154b3c4d..09da48351ef 100644 --- a/.github/workflows/_unit_test_coverage.yml +++ b/.github/workflows/_unit_test_coverage.yml @@ -41,7 +41,7 @@ jobs: run_tests_with_coverage: runs-on: [self-hosted, GPU-h1z1-2Cards] - timeout-minutes: 60 + timeout-minutes: 90 needs: check_cov_skip if: needs.check_cov_skip.outputs.can-skip != 'true' outputs: diff --git a/tools/dockerfile/Dockerfile.ci b/tools/dockerfile/Dockerfile.ci index 32fb95e0180..6d711889f0d 100644 --- a/tools/dockerfile/Dockerfile.ci +++ b/tools/dockerfile/Dockerfile.ci @@ -1,6 +1,9 @@ FROM ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddleqa:cuda126-py310-cibase RUN apt update && apt install -y jq +RUN python -m pip install --no-cache-dir xgrammar==0.1.19 torch==2.6.0 -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple COPY ../../requirements.txt ./requirements.txt -RUN python -m pip install -r requirements.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple && python -m pip install pytest +RUN python -m pip install --no-cache-dir -r requirements.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple COPY ./unittest_requirement.txt ./unittest_requirement.txt -RUN python -m pip install -r unittest_requirement.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple +RUN python -m pip install --no-cache-dir -r unittest_requirement.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple +COPY ./requirements_paddle_nv.txt ./requirements_paddle_nv.txt +RUN python -m pip install --no-cache-dir -r requirements_paddle_nv.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple diff --git a/tools/dockerfile/requirements_paddle_nv.txt b/tools/dockerfile/requirements_paddle_nv.txt new file mode 100644 index 00000000000..d31402b169f --- /dev/null +++ b/tools/dockerfile/requirements_paddle_nv.txt @@ -0,0 +1,14 @@ +nvidia-cuda-nvrtc-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-cuda-runtime-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-cuda-cupti-cu12==12.6.80; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-cudnn-cu12==9.5.1.17; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-cublas-cu12==12.6.4.1; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-cufft-cu12==11.3.0.4; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-curand-cu12==10.3.7.77; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-cusolver-cu12==11.7.1.2; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-cusparse-cu12==12.5.4.2; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-cusparselt-cu12==0.6.3; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-nccl-cu12==2.25.1; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-nvtx-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-nvjitlink-cu12==12.6.85; platform_system == 'Linux' and platform_machine == 'x86_64' +nvidia-cufile-cu12==1.11.1.6; platform_system == 'Linux' and platform_machine == 'x86_64'