Update pypi-distribution-reusable.yaml

ewanwm · Sep 23, 2024 · e94f0de · e94f0de
1 parent d932fd2
commit e94f0de
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.github/workflows/pypi-distribution-reusable.yaml b/.github/workflows/pypi-distribution-reusable.yaml
@@ -60,7 +60,10 @@ jobs:
           ls -al /usr/local &&
           export PATH=$PATH:/usr/local/cuda/bin &&
           nvcc --version &&
-          export BUILD_WITH_CUDA=1
+          echo nvcc location: &&
+          which nvcc &&
+          export CUDACXX="$(which nvcc)"
+          export BUILD_WITH_CUDA=1 &&
           export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0"
       
     #- name: Build wheels