-
Notifications
You must be signed in to change notification settings - Fork 25.7k
Closed
Labels
module: buildBuild system issuesBuild system issuesmodule: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generaltriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
🐛 Describe the bug
Fix 9.0a not being recognized as a CUDA architecture
CMake Error at /usr/local/lib/python3.10/dist-packages/cmake/data/share/cmake-3.28/Modules/CMakeDetermineCUDACompiler.cmake:270 (message):
Failed to detect a default CUDA architecture.
Compiler output:
Call Stack (most recent call first):
cmake/public/cuda.cmake:47 (enable_language)
cmake/Dependencies.cmake:44 (include)
CMakeLists.txt:753 (include)
and
CMake Error at cmake/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake:225 (message):
Unknown CUDA Architecture Name 9.0a in CUDA_SELECT_NVCC_ARCH_FLAGS
Call Stack (most recent call first):
cmake/public/utils.cmake:401 (cuda_select_nvcc_arch_flags)
cmake/public/cuda.cmake:342 (torch_cuda_get_nvcc_gencode_flag)
cmake/Dependencies.cmake:44 (include)
CMakeLists.txt:753 (include)
CMake Error at cmake/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake:229 (message):
arch_bin wasn't set for some reason
Call Stack (most recent call first):
cmake/public/utils.cmake:401 (cuda_select_nvcc_arch_flags)
cmake/public/cuda.cmake:342 (torch_cuda_get_nvcc_gencode_flag)
cmake/Dependencies.cmake:44 (include)
CMakeLists.txt:753 (include)
At our org we use a modified version of PyTorch and I hadn't experienced this yet.
Surprised this is an issue since xformers has already implemented it
facebookresearch/xformers@ad1046f
Versions
Nightly
Metadata
Metadata
Assignees
Labels
module: buildBuild system issuesBuild system issuesmodule: cudaRelated to torch.cuda, and CUDA support in generalRelated to torch.cuda, and CUDA support in generaltriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module