问题描述:
使用神经网络搭建一个简单的分类器,运行代码时,
错:“tensorflow.python.framework.errors_impl.InternalError: Blas GEMM launch failed”
提示:查阅相关资料,怀疑是tensorflow的版本问题
当前配置如下:
- tensorflow-gpu:1.15.0
- cuda:10.0
- cudnn:7.6.5
- python:3.6
解决方案:
重置tensorflow环境,更新版本如下:
- tensorflow-gpu:2.1.0
- cuda:10.1
- cudnn:7.6.0
- python:3.6
完美运行代码!
TF的坑太多了,只能遇到一个,解决一个!