无法创建cudnn句柄:CUBLAS_STATUS_ALLOC_FAILED

问题描述 投票:1回答:1

我正在尝试运行我在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

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

python tensorflow keras gpu cublas
1个回答
1
投票

我发现它来自哪里。实际上,我在环境变量中添加了两个不同的路径(一个用于cuda 8,另一个用于cuda 9)。删除cuda 8删除此错误!

© www.soinside.com 2019 - 2024. All rights reserved.