From 6798e99f4f3b12e4949f02d165f64ca383f4eda8 Mon Sep 17 00:00:00 2001 From: Lu Weizheng Date: Mon, 19 Aug 2024 18:58:57 +0800 Subject: [PATCH] CI GPU --- .github/workflows/python.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/python.yaml b/.github/workflows/python.yaml index 3cf2ce2..9677014 100644 --- a/.github/workflows/python.yaml +++ b/.github/workflows/python.yaml @@ -87,7 +87,7 @@ jobs: - { os: windows-latest, python-version: 3.9} - { os: windows-latest, python-version: 3.10} include: - - { os: self-hosted, module: gpu, python-version: 3.9} + - { os: self-hosted, module: gpu, python-version: 3.11} steps: - name: Check out code @@ -135,6 +135,7 @@ jobs: run: | ${{ env.SELF_HOST_PYTHON }} -m pip install --extra-index-url=https://pypi.nvidia.com cudf-cu12 ${{ env.SELF_HOST_PYTHON }} -m pip install ucx-py-cu12 cloudpickle psutil tblib uvloop packaging "numpy<2.0.0" scipy cython coverage flaky + ${{ env.SELF_HOST_PYTHON }} setup.py clean --all ${{ env.SELF_HOST_PYTHON }} -m pip install -e ".[dev,extra]" ${{ env.SELF_HOST_PYTHON }} setup.py build_ext -i working-directory: ./python