在pycharm tensorflow-gup运行时报错Function call stack: xxxxxx_function的解决办法
程序员文章站
2022-05-10 17:14:44
...
方法一:
在代码前加上以下语句:
import tensorflow as tf
config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
sess = tf.compat.v1.Session(config=config)
【注意:tensorflow1.x版本跟2.x版本的语法有区别,我这里是在2.x下的语句。】
方法二:
import tensorflow as tf
physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
tf.config.experimental.set_memory_growth(physical_devices[0], True)
【注意:
tf.config.experimental.set_memory_growth(physical_devices[i], True)中的i取决于设备的gpu数量。就是自己电脑上有多少gpu这条语句就调用多少次啦!】