问题:
最近在复现Mask_RCNN代码时,报错如下:
原因:
cuda 与 cudnn 未正确配置
解决方法:
需要到官网手动下载cuda与cudnn:
cuda官网:CUDA Toolkit Archive | NVIDIA Developer
cudnn官网:cuDNN Archive | NVIDIA Developer
注意:cuda 与 cudnn的版本号需要与tensorflow-gpu的版本号相对应。
官网链接:在 Windows 环境中从源代码构建 | TensorFlow (google.cn)
作者本人使用的是tensorflow-gpu 1.3.0 ,所以下载了cuda 8.0 GA1, cuDNN v5(年份相同)。
运行结果(不再报之前的错误,成功):