- 博客(2)
- 收藏
- 关注
RSS订阅原创 tensorflow.python.framework.errors_impl.InternalError: Blas GEMM launch failed
此错误主要是GPU的可用内存不足引起的错误，解决方法如下： import tensorflow as tf import os os.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
4261
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人