Skip to content

Commit

Permalink
Update pypi-distribution-reusable.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
ewanwm authored Sep 24, 2024
1 parent 7236358 commit b344de2
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions .github/workflows/pypi-distribution-reusable.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit b344de2

Please sign in to comment.