TensorFlow-GPU版本,载入模型时报错:Blas GEMM。。解决办法
如果安装了GPU版本的TensorFlow,使用其进行模型的载入时,遇到报错,类似下面这种:
InternalError: Blas GEMM launch failed : a.shape=(10000, 784), b.shape=(784, 10), m=10000, n=10, k=784
[[node MatMul (defined at <ipython-input-2-03...
原创
2018-12-28 21:06:26 ·
4866 阅读 ·
3 评论