欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

在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这条语句就调用多少次啦!】

相关标签: 笔记 tensorflow