我使用numba库中的装饰器@jit优化了一些python代码。但是,我想指示@jit显式地使用我的GPU设备。出发地:,我知道我需要使用@jit(target="cuda")来完成它。我试着这样做:
@jit(target='cuda') # The code runs normally without (target='cudano_cfu
funniest /__init__.py", line 2, in <module>File "/home/domi89/Dropbox/Python_Projectsfunniest/funniest /ModuleA.py", line 1, in <module>File "/home/domi89/
对装饰器进行更好的动态控制的最佳方法是从numba.cuda.jit、numba.jit和none (纯python)中进行选择。请注意,一个项目可以有10或100的功能,所以这应该是很容易应用到所有的功能,这里是一个例子,从numba网站。import numba as nb
# global control of this --> @nb.jit or @nb.cuda.jit or none# s
我正在遵循https://fenics.readthedocs.io/projects/containers/en/latest/introduction.html的指示。我得到了: fenics@af86d2c8af46:~$ cd ~/demo/python/documented/poisson
fenics@af86d2c8af46:~/demo/python/documented/po