YOLO v3 tensorflow inference报错failed to run cuBLAS routine cublasSgemm_v2: CUBLAS_STATUS_EXECUTION_FAILED的解决办法

YOLO v3 tensorflow inference 报错failed to run cuBLAS routine cublasSgemm_v2: CUBLAS_STATUS_EXECUTION_FAILED的解决办法

环境cuda9.0+cudnn7.1.3+tensorflow1.8.0
显卡 rtx2080ti

原因是NVIDIA cuda 9.0自身的bug
解决方法安装cuda9.0的四个patch，文件分别是cuda9.0.176.1_linux.run, cuda9.0.176.2_linux.run,cuda9.0.176.3_linux.run,cuda9.0.176.4_linux.run 在NVIDIA官网上就能下载，不过速度慢

不方便的可以直接用下面百度网盘的地址

链接: https://pan.baidu.com/s/1Y3wrq6GWHtNwZWJ5bz8tJg 密码: ep4u

下载完成后 直接运行安装patch

1
2
3
4
sudo sh cuda9.0.176.1_linux.run

sudo sh cuda9.0.176.2_linux.run

sudo sh cuda9.0.176.3_linux.run

sudo sh cuda9.0.176.4_linux.run

按提示安装完毕，再运行就不会报错了