KEMBAR78
Fix 9.0a not being recognized as a CUDA architecture · Issue #119946 · pytorch/pytorch · GitHub
Skip to content

Fix 9.0a not being recognized as a CUDA architecture #119946

@ConsceIeratus

Description

@ConsceIeratus

🐛 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

cc @malfet @seemethere @ptrblck

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: buildBuild system issuesmodule: cudaRelated to torch.cuda, and CUDA support in generaltriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions