diff --git a/.github/workflows/requirements_cuda_ci.txt b/.github/workflows/requirements_cuda_ci.txt index ef35e340..da9fde80 100644 --- a/.github/workflows/requirements_cuda_ci.txt +++ b/.github/workflows/requirements_cuda_ci.txt @@ -1,6 +1,4 @@ numpy==2.2.5 torch==2.7.0 --index-url https://download.pytorch.org/whl/cu128 pytest==8.3.5 -ninja==1.11.1.4 -ruff==0.11.11 -pre-commit==4.2.0 \ No newline at end of file +ninja==1.11.1.4 \ No newline at end of file diff --git a/.github/workflows/verify_extension_build.yml b/.github/workflows/verify_extension_build.yml index db48af7b..83f1e894 100644 --- a/.github/workflows/verify_extension_build.yml +++ b/.github/workflows/verify_extension_build.yml @@ -13,8 +13,10 @@ permissions: jobs: verify_cuda_extension: if: ${{ github.event.label.name == 'ci-ready' || github.event_name != 'pull_request' }} - runs-on: ubuntu-latest - + runs-on: ubuntu-24.04 + + container: + image: nvcr.io/nvidia/cuda:12.9.0-devel-ubuntu24.04 # Pre installed CUDA toolkit steps: - uses: actions/checkout@v4 - name: Set up Python @@ -26,8 +28,6 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip - sudo apt-get update - sudo apt install nvidia-cuda-toolkit pip install -r .github/workflows/requirements_cuda_ci.txt pip install -e .