python - 无法创建 cudnn 句柄：CUBLAS_STATUS_ALLOC_FAILED

Question

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

一切正常。

但是，不幸的是，我收到了这个奇怪的错误，我在互联网上找不到任何解决方案：'could not create cudnn handle: CUBLAS_STATUS_ALLOC_FAILED

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

标签： pythontensorflowkerasgpucublas

Answer 1

我发现它来自哪里。实际上，我在环境变量中添加了两条不同的路径（一条用于 cuda 8，一条用于 cuda 9）。删除 cuda 8 的删除此错误！