diff --git a/.github/actions/setup-env-build/action.yml b/.github/actions/setup-env-build/action.yml
index edbc9872..6db5163a 100644
--- 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 .