怎么知道后端是否成功使用了gpu

综合讨论区
#1

我在使用gpu后端运行时发现资源大多是cpu占用，gpu很少几乎没有使用
这是怎么回事呀，如何能够验证后端使用的是什么

#2

可以试试看 print(ti.cfg.arch)。有时即使是GPU后端，还是会有（单核）CPU高消耗，似乎是内存拷贝导致的。

#3

噢噢谢谢!

#4

print(ti.cfg.arch) 显示是 Arch.x64

应该是 gpu没启用成功吧

我的确装了 cuda，想要用上，怎么查找没成功运行的原因呢

cuda 10.2
Python 3.7.3
taichi 0.6.7

#5

你试试看python -m taichi:

[Taichi] mode=release
[Taichi] version 0.6.7, supported archs: [cpu, cuda, opengl], commit ca4d9dda, python 3.8.2

看看是不是 [cpu, cuda, opengl]？

还有，你需要在代码里指定 ti.init(arch=ti.gpu) 才能使用 GPU，否则默认是 CPU 的。

#6

哦哦没问题了，是可以跑的，谢谢啦
python .\waterwave.py
[Taichi] mode=release
[Taichi] version 0.6.7, supported archs: [cpu, cuda, opengl], commit ca4d9dda, python 3.7.3
Arch.cuda
[Hint] click on the window to create wavelet