thu-ml / thu-ml/TurboDiffusion

报错 nvcc fatal : Unsupported gpu architecture 'compute_120a'

Open
#80 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
3.7k
Forks
277
Avg merge
2h 57m
Merged PRs (30d)
2

Description

我的显卡是 Nvidia A10 24G,通过 conda create 进行部署,执行最后一条命令 pip install turbodiffusion --no-build-isolation 后,运行报错。
conda create -n turbodiffusion python=3.12
conda activate turbodiffusion
pip install turbodiffusion --no-build-isolation

报错信息为:
DCUTLASS_DEBUG_TRACE_LEVEL=0 -DNDEBUG -Xcompiler -fPIC -DEXECMODE=0 -gencode arch=compute_120a,code=sm_120a -gencode arch=compute_90,code=sm_90 -gencode arch=compute_89,code=sm_89 -gencode arch=compute_80,code=sm_80 --threads 4 -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=turbo_diffusion_ops -D_GLIBCXX_USE_CXX11_ABI=0
nvcc fatal : Unsupported gpu architecture 'compute_120a'

既然宣传的 RTX-5090 都能运行,为什么会出现 nvcc fatal : Unsupported gpu architecture 'compute_120a' 的提示呢,我修改了
./ops/cutlass/CMakeLists.txt: list(APPEND CUTLASS_NVCC_ARCHS_SUPPORTED 100 100a 120 120a 121 121a)
./ops/cutlass/customConfigs.cmake: set(PROFILER_ARCH_LIST 100a 100f 103a 120a 120f 121a)

修改为了 80,89和90, 是应该这么修改吗,如何解决上面的报错。

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reproducing the final pip install turbodiffusion --no-build-isolation command in the reported conda environment and inspect the nvcc architecture error. Read ops/cutlass/CMakeLists.txt and ops/cutlass/customConfigs.cmake, then verify which architectures the installed nvcc supports. Done means installation completes for the Nvidia A10 without the compute_120a error.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
build-system, machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.