- 博客(2)
- 收藏
- 关注
RSS订阅原创 tensorflow.python.framework.errors_impl.InternalError: Blas GEMM launch failed
此错误主要是GPU的可用内存不足引起的错误，解决方法如下：import tensorflow as tfimport osos.environ["CUDA_VISIBLE_DEVICES"] = '0' #或者'1' 调用运行GPU的编号# 定义TensorFlow配置config = tf.ConfigProto()# 配置GPU内存分配方式，按需增长，很关键config.gpu_options.allow_growth = True# 配置可使用的显存比例config.
2022-04-14 20:21:10
3879
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人