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