Taichi on NVIDIA A100 GPU

taichi-dev / taichi

Productive, portable, and performant GPU programming in Python.

https://taichi-lang.org

Apache License 2.0

25.41k stars 2.27k forks source link

Open tomhitu opened 8 months ago

tomhitu commented 8 months ago

Describe the bug [misc.py:adaptive_arch_select@758] Arch=[<Arch.cuda: 3>] is not supported, falling back to CPU

To Reproduce


import taichi as ti
ti.init(arch=ti.cuda, debug=False)

$ python my_sample_code.py

[Taichi] version 1.7.0, llvm 15.0.4, commit 2fd24490, linux, python 3.9.13 ...

bobcao3 commented 7 months ago

Hi, can you check whether you have installed it from the manylinux compatibility version that only has the CPU backend?