Update pypi-distribution-reusable.yaml

ewanwm · Sep 24, 2024 · fb663ce · fb663ce
1 parent be39a5b
commit fb663ce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/workflows/pypi-distribution-reusable.yaml b/.github/workflows/pypi-distribution-reusable.yaml
@@ -68,12 +68,12 @@ 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])')/torch/lib/ 
+          PATH=$PATH:/usr/local/cuda/bin LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$(python -c 'import sys; print(sys.prefix + "/lib/python" + sys.version[:3] + "/site-packages")')/torch/lib/ 
         CIBW_BEFORE_BUILD: >
           python -m site &&
-          python -c 'import site; print(site.getsitepackages())' &&
+          python -c 'import sys; print(sys.prefix + "/lib/python" + sys.version[:3] + "/site-packages")' &&
           echo $LD_LIBRARY_PATH &&
-          ls $(python -c 'import site; print(site.getsitepackages()[0])')
+          ls $(python -c 'import sys; print(sys.prefix + "/lib/python" + sys.version[:3] + "/site-packages")')
         
       
     #- name: Build wheels