Repository navigation
[FEATURE]: Allow cuda build on host without a device when TORCH_CUDA_ARCH_LIST is set #5300
Description
Describe the feature
When TORCH_CUDA_ARCH_LIST is set, allow gpu build to succeed by not searching for a device and resulting in
RuntimeError: No CUDA GPUs are available
when the host does not have a device.
This is especially important for HPC centre where build nodes may not have a GPU device, and where we build for multiple architectures.
Thanks!!
Activity
JazyJiang commentedon Oct 22, 2026
Closing as completed. When no GPU is visible, the CUDA extension build now respects
TORCH_CUDA_ARCH_LIST, and picks a default list from the CUDA version if it is unset (#6459). SoBUILD_EXT=1 pip install .works on a GPU-less node. This is onmainand will be in the next release.