diff --git a/.github/workflows/pypi-distribution-reusable.yaml b/.github/workflows/pypi-distribution-reusable.yaml index 9172bb8..2a2fa56 100644 --- a/.github/workflows/pypi-distribution-reusable.yaml +++ b/.github/workflows/pypi-distribution-reusable.yaml @@ -68,6 +68,7 @@ jobs: export BUILD_WITH_CUDA=1 && export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0" CIBW_ENVIRONMENT: PATH=$PATH:/usr/local/cuda/bin + CIBW_BEFORE_BUILD: python -m site #- name: Build wheels