安装Anaconda加速器/NumbaPro CUDA在OSX上链接错误

5

总体目标是使用NumbaPro在GPU上运行一些函数(在OSX 10.8.3上)。

在开始之前,我只是想把一切都准备好。根据this page的说明,我安装了CUDA,注册为CUDA开发人员,下载了编译器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而不是autojit,但是出现了相同的错误。 仅使用@autojit而不指定目标时可以正常工作。 有任何想法吗?

1
你也可以询问NumbaPro人员。我注意到在你提供的网站上,它声明:“GPU支持正在迅速提高,但仍然可能会遇到困难。如果您在使用我们的GPU支持时遇到任何问题,请告诉我们。”作为一个产品,在30天后需要支付使用费用,我认为他们应该提供某种形式的支持。 - Robert Crovella
这实际上并不是一个CUDA问题。我很想完全删除CUDA标签... - talonmies
@RobertCrovella,我知道NumbaPro是一个付费产品,我正在使用学术许可证。我会尝试给他们发送电子邮件,但我希望能够找到一个快速简便的解决方案。 - thearrow3456
@talonmies,你是对的。完成了。 - thearrow3456
1个回答

5

为了后世留存,我向Continuum Support提出了问题。他们回答说:

看起来您正在使用具有计算能力1.x的CUDA GPU。NVVM仅支持CC2.0及以上版本。我们一定要有更好的错误报告,并在NumbaPro文档中明确支持的计算能力。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接