Update pypi-distribution-reusable.yaml

ewanwm · Sep 24, 2024 · b1ca2d2 · b1ca2d2
1 parent b344de2
commit b1ca2d2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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