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 921309d commit 70d7d54
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/pypi-distribution-reusable.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ jobs:
echo gcc version: &&
gcc --version &&
export BUILD_WITH_CUDA=1 &&
export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0" &&
export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0"
CIBW_ENVIRONMENT: >
PATH=$PATH:/usr/local/cuda/bin LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$(python -c 'import site; print(site.getsitepackages()[0])')
CIBW_BEFORE_BUILD: >
Expand Down

0 comments on commit 70d7d54

Please sign in to comment.