failed to run cuBLAS routine cublasSgemm_v2: CUBLAS_STATUS_EXECUTION_FAILED解决方案
服务器运行tensorflow gpu程序报错:failed to run cuBLAS routine cublasSgemm_v2: CUBLAS_STATUS_EXECUTION_FAILED解决方案
首先 查看cuda版本和tensorflow-gpu版本是否对应,不对应的话更改tensorflow-gpu版本
其次 在import tensorflow as tf这一句下面,注意不可以是别的地方,添加如下代码
# add to the top of your code under impo
原创
2021-04-16 10:52:41 ·
8358 阅读 ·
0 评论