From b1ca2d259aaa62b1e16439e4836b0fbf404dc1d2 Mon Sep 17 00:00:00 2001 From: Ewan Miller Date: Tue, 24 Sep 2024 09:04:12 -0500 Subject: [PATCH] Update pypi-distribution-reusable.yaml --- .github/workflows/pypi-distribution-reusable.yaml | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/.github/workflows/pypi-distribution-reusable.yaml b/.github/workflows/pypi-distribution-reusable.yaml index 2a2fa56..8198b88 100644 --- a/.github/workflows/pypi-distribution-reusable.yaml +++ b/.github/workflows/pypi-distribution-reusable.yaml @@ -66,8 +66,11 @@ jobs: echo gcc version: && gcc --version && 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 + export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0" && + python -c 'import site; print(site.getsitepackages()) + 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: python -m site