CUDA_HOME

facebookresearch · Oct 29, 2024 · b3533d7 · b3533d7
1 parent e596bcf
commit b3533d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/actions/setup-env-build/action.yml b/.github/actions/setup-env-build/action.yml
@@ -146,4 +146,4 @@ runs:
     - name: Build
       shell: bash -l {0}
       run: |
-        FORCE_CUDA=1 PYTORCH_NVCC="$CONDA_PREFIX/bin/nvcc-ccache" CXX="g++-ccache" TORCH_CUDA_ARCH_LIST=${{ inputs.arch }} python -m pip install -v -e .
+        CUDA_HOME=$CONDA_PREFIX PYTORCH_NVCC="$CONDA_PREFIX/bin/nvcc-ccache" CXX="g++-ccache" TORCH_CUDA_ARCH_LIST=${{ inputs.arch }} python -m pip install -v -e .