gpt4 book ai didi

python - 无法创建 cudnn 句柄 : CUBLAS_STATUS_ALLOC_FAILED

转载 作者:行者123 更新时间:2023-12-01 09:07:13 25 4
gpt4 key购买 nike

我正在尝试运行我在 GitHub 上找到的这个气球分割模型: https://github.com/matterport/Mask_RCNN/tree/master/samples/balloon.

它在我的 Linux 虚拟机上运行,​​但非常非常慢,所以我在几次迭代后停止它,并在我的本地 Windows 操作系统上安装了tensorflow-gpu,使用: https://www.codingforentrepreneurs.com/blog/install-tensorflow-gpu-windows-cuda-cudnn/

一切正常。

但是,不幸的是,我遇到了这个奇怪的错误,我在互联网上找不到任何解决方案:'无法创建 cudnn 句柄:CUBLAS_STATUS_ALLOC_FAILED

任何建议都会有很大帮助!我现在被封锁两天了,不知道现在该尝试什么......

最佳答案

我找到了它的来源。我实际上在环境变量中添加了两个不同的路径(一个用于 cuda 8,一个用于 cuda 9)。删除 cuda 8 的那个即可消除此错误!

关于python - 无法创建 cudnn 句柄 : CUBLAS_STATUS_ALLOC_FAILED,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51951805/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com