我试图通过CUDA @vectorize一些python。我用Cuda9.1,Anaconda3在原子中运行Python3,并且遇到类似的问题:
但是这个解决方案适用于Linux而且我在Windows上。
我从上面的解决方案中收集我需要添加一个环境变量。我试着通过从我找到的nvvm文件中搜索来做到这一点:
C:\用户\ ME \应用程序数据\本地\程序\的Python \ Python36 \ LIB \站点包\ numba \ CUDA \测试\ cudadrv__pycache __
但补充说没有帮助。 我试过'conda install cudatoolkit'以及更新它。
我的问题是操作系统不知道这个库在哪里?如何将python指向它需要的库?
答案 0 :(得分:6)
这些是我设置的环境变量,用于在Windows上运行一些pyCUDA示例代码:
NUMBAPRO_NVVM = C:\ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v9.1 \ nvvm \ bin \ nvvm64_32_0.dll
NUMBAPRO_LIBDEVICE = C:\ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v9.1 \ nvvm \ libdevice \
答案 1 :(得分:2)
您必须从最后删除栏。
NUMBAPRO_LIBDEVICE = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\nvvm\libdevice
答案 2 :(得分:2)
Windows 10(x64)上的Python 3.7.0(AMD上为x64)。 CUDA 10
pip install --upgrade pip
pip install numba
pip install numpy
pip install six
还需要将系统变量路径设置为:
NUMBAPRO_NVVM=c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\nvvm\bin\nvvm64_33_0.dll
NUMBAPRO_LIBDEVICE=c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\nvvm\libdevice
重新启动计算机!(由于某些原因,这似乎很重要)。 这对我有用。
答案 3 :(得分:2)