diff --git a/.github/workflows/pypi-distribution-reusable.yaml b/.github/workflows/pypi-distribution-reusable.yaml index d28c70c..64b7e21 100644 --- a/.github/workflows/pypi-distribution-reusable.yaml +++ b/.github/workflows/pypi-distribution-reusable.yaml @@ -68,7 +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 LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$(python -c 'import site; print(site.getsitepackages()[0])') + PATH=$PATH:/usr/local/cuda/bin LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$(python -c 'import site; print(site.getsitepackages()[0])')/torch/lib/ CIBW_BEFORE_BUILD: > python -m site && python -c 'import site; print(site.getsitepackages())'