首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >PyCUDA --导入pycuda.driver的问题

PyCUDA --导入pycuda.driver的问题
EN

Stack Overflow用户
提问于 2022-03-01 23:14:45
回答 1查看 231关注 0票数 1
  • Windows 10
  • Python3.8
  • CUDA 11.5

我从这个文件中安装了一个我认为是匹配的pycuda :pycuda-2021.1+pycuda 115-cp38-cp38-win_a 64。

这个简单的例子失败了

代码语言:javascript
运行
复制
import pycuda.driver as drv

drv.init()

print("Detected {} CUDA devices".format(drv.Device.count()))

有了这个错误:

代码语言:javascript
运行
复制
Traceback (most recent call last):
  File "C:/University of Arizona/weeds/tests/cuda-summary.py", line 5, in <module>
    import pycuda.driver as drv
  File "C:\Users\evan\AppData\Local\Programs\Python\Python38\lib\site-packages\pycuda\driver.py", line 65, in <module>
    from pycuda._driver import *  # noqa
ImportError: DLL load failed while importing _driver: The specified procedure could not be found.

os.add_dll_directory(os.path.join(os.environ'CUDA_PATH',

  • NVCC在我的路径中

  • 添加

'bin'))对没有影响

剧本在我的Jetson Nano上运行得很好

有什么办法克服这件事吗?我已经搜索并尝试了几种解决方案。

EN

回答 1

Stack Overflow用户

发布于 2022-06-06 19:45:07

这是相当奇怪的,但更新的Nvidia显示驱动程序为我解决了这个问题。

但在我的例子中,我的个人电脑上没有问题,与Windows、Python、CUDA和PyCuda的版本相同。这个问题出现在不同的Windows 10机器上,当启动由PyInstaller打包的exe时。因此,更新显示驱动程序是此机器的修复程序。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/71315681

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档