时间戳

首页 > 解决方案 > 无法在 Python 应用程序中设置 cuSparse 环境变量

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

问题描述

我第一次尝试使用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/SI 以前libnvvm通过以下方式成功地为 Numba 设置了一个环境变量：

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

但似乎同样的技巧在这里不起作用。

标签： pythonenvironment-variables

解决方案

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

英伟达开发者对话

这是给我的

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

推荐阅读