Tensorflow2中Blas GEMM launch failed/CUBLAS_STATUS_NOT_INITIALIZED错误

Blas GEMM launch failed/CUBLAS_STATUS_NOT_INITIALIZED错误

问题分析

Tensorflow2中Blas GEMM launch failed/CUBLAS_STATUS_NOT_INITIALIZED错误
Tensorflow2中Blas GEMM launch failed/CUBLAS_STATUS_NOT_INITIALIZED错误
问题现象如图, 去tensorflow的issue区查看看了下问题,给出的解决方案大多数是说你应该重启下程序或者是确保其他程序没有占用gpu和设置gpu使用

但是实际问题是出在了libcublas上,因为目前的libcublas与cuda10-1不匹配

解决方法

修改了libcublas到相应匹配版本即可

Tensorflow2中Blas GEMM launch failed/CUBLAS_STATUS_NOT_INITIALIZED错误
重定向软连接后
Tensorflow2中Blas GEMM launch failed/CUBLAS_STATUS_NOT_INITIALIZED错误
然后tensorflow的程序就可以正常运行了。

总结

问题出在了libcublas与cuda版本不匹配,降级一下就可以了。