From b344de295b192426de8435c5a2bade68eeb7a337 Mon Sep 17 00:00:00 2001 From: Ewan Miller Date: Tue, 24 Sep 2024 08:37:25 -0500 Subject: [PATCH] Update pypi-distribution-reusable.yaml --- .github/workflows/pypi-distribution-reusable.yaml | 1 + 1 file changed, 1 insertion(+) 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