无法创建cudnn句柄：CUBLAS_STATUS_ALLOC_FAILED

Posted 2021-04-30

我正在尝试运行我在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删除此错误！