项目场景:
TF Version: 2.4.0-dev20201015
TF-Hub version: 0.9.0
Eager mode enabled: True
问题描述:
InternalError: Blas SGEMM launch failed
解决方案:
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
for gpu in gpus:
tf.config.experimental.set_memory_growth(device=gpu, enable=True)
第二个启用此选项的方式是将环境变量 TF_FORCE_GPU_ALLOW_GROWTH 设置为 true。这是一个特定于平台的配置。
pycharm可以设置环境变量