CI GPU

xorbitsai · Aug 19, 2024 · 6798e99 · 6798e99
1 parent dc51dac
commit 6798e99
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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