Manually specify flags if no arch set

stack-info: PR: #2219, branch: drisspg/stack/55

Author: drisspg

CUDA_ARCH_NOTES.md

@@ -79,6 +79,24 @@ def use_debug_mode():
     _get_cuda_arch_flags,
 )
 
+# =====================================================================================
+# CUDA Architecture Settings
+# =====================================================================================
+# If TORCH_CUDA_ARCH_LIST is not set during compilation, PyTorch tries to automatically
+# detect architectures from available GPUs. This can fail when:
+#   1. No GPU is visible to PyTorch
+#   2. CUDA is available but no device is detected
+#
+# To resolve this, you can manually set CUDA architecture targets:
+#   export TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6+PTX"
+#
+# Adding "+PTX" to the last architecture enables JIT compilation for future GPUs.
+# =====================================================================================
+if "TORCH_CUDA_ARCH_LIST" not in os.environ and torch.version.cuda:
+    # Set to common architectures for CUDA 12.x compatibility
+    os.environ["TORCH_CUDA_ARCH_LIST"] = "7.0;7.5;8.0;8.6;8.9;9.0,10.0+PTX"
+    print(f"Setting default TORCH_CUDA_ARCH_LIST={os.environ['TORCH_CUDA_ARCH_LIST']}")
+
 IS_ROCM = (torch.version.hip is not None) and (ROCM_HOME is not None)