Update pypi-distribution-reusable.yaml

ewanwm · Sep 24, 2024 · 138eb47 · 138eb47
1 parent 70d7d54
commit 138eb47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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())'