无法在Python应用程序中设置cuSparse环境变量

Question

我第一次尝试使用pyculib。我做了pyculib：

pip install PyCulib

但是当我尝试一个简单的导入语句时，如：

import pyculib

我收到错误消息：

Exception: Cannot open library for cusparse: library cusparse not found

谷歌搜索一下，我认为这是因为cuSPARSE库没有链接到我的Python应用程序。因此，我决定尝试通过设置环境变量来链接它：

import os
os.environ['PYCULIB_CUSPARSE']='/usr/local/cuda-9.0/lib64/libcusparse.so'

为了确保设置正确，我检查了两件事：

1. 确保cusparse库的路径正确

我使用命令locate cusparse来查找路径。

1. 确保已设置变量

我在同一个应用程序中设置变量后立即执行了print(os.environ.get('PYCULIB_CUSPARSE'))，以检查变量是否确实已设置。

尽管如此，当我再次尝试import pyculib时（在同一个应用程序中），会出现相同的错误消息，警告未找到库cusparse。知道发生了什么，以及如何在Python应用程序中进行正确的链接？

P / S我之前通过以下方式设法为Numba libnvvm设置了一个环境变量（成功）：

os.environ['NUMBAPRO_NVVM']='/usr/local/cuda-9.0/nvvm/lib64/libnvvm.so'
os.environ['NUMBAPRO_LIBDEVICE']='/usr/local/cuda-9.0/nvvm/libdevice/'

但似乎同样的伎俩在这里不起作用。

Answer 1

我建议你试试

conda install pyculib

这是Pyculib自己推荐的方法：

http://pyculib.readthedocs.io/en/latest/install.html#installing-pyculib

Answer 2

NVIDIA说要在NUMBAPRO_CUDALIB变量中设置CUDA / bin

nvidia devtalk

这是给我的

import os
os.environ['NUMBAPRO_CUDALIB']='C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1\\bin'