无法创建cudnn句柄：CUBLAS_STATUS_ALLOC_FAILED

Question

我正在尝试运行在GitHub上找到的此Ballon细分模型： 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/

一切正常。

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

任何建议都会有很大帮助！我两天以来就被封锁了，不知道该怎么办...

Answer 1

我找到它的来源。实际上，我在环境变量中添加了两条不同的路径（一条用于cuda 8，一条用于cuda 9）。删除cuda 8的那个消除此错误！