python – Anaconda Acclerate / NumbaPro CUDA链接错误OSX

总体目标是使用NumbaPro在GPU上运行某些功能(在OSX 10.8.3上).

在开始之前,我只想设置一切.根据this page我安装了CUDA,注册为CUDA开发人员,下载了Compiler SDK并设置了NUMBAPRO_NVVM = / path / to / libnvvm.dylib环境变量.

但是,运行此基本测试功能：

from numbapro import autojit

@autojit(target='gpu')
def my_function(x):
    if x == 0.0:
        return 1.0
    else:
        return x*x*x

print my_function(4.4)
exit()

出现此错误：

File ".../anaconda/lib/python2.7/site-packages/numba/decorators.py", line 207, in compile_function
    compiled_function = dec(f)
File "...lib/python2.7/site-packages/numbapro/cudapipeline/decorators.py", line 35, in _jit_decorator
File "...lib/python2.7/site-packages/numbapro/cudapipeline/decorators.py", line 128, in __init__
File "...lib/python2.7/site-packages/numbapro/cudapipeline/environment.py", line 31, in generate_ptx
File "...lib/python2.7/site-packages/numbapro/cudapipeline/environment.py", line 186, in _link_llvm_math_intrinsics
KeyError: 1

我试过@ vectorize’ing而不是autojit,同样的错误.
@autojit本身没有目标工作正常.

有任何想法吗？

解决方法:

为了后人的缘故,我问Continuum Support.他们回答说：

It seems that you are running a CUDA GPU with compute capability 1.x. NVVM only supports CC2.0 and above. We definitely should have a better error reporting and make it clear in the NumbaPro documentation for the supported compute capability.

标签：python,jit,gpu,anaconda,numba-pro
来源： https://codeday.me/bug/20190520/1142965.html