小编the*_*eff的帖子

如何使用tensorflow-gpu GPUOptions

我正在尝试使用tensorflow-gpu,这样我就可以更快地运行我的模型,但我真的不明白如何激活它。目前我的代码如下所示:

import tensorflow as tf

gpu_options = tf.compat.v1.GPUOptions(set_per_process_memory_fraction(0.333))
sess = tf.Session(config=tf.compat.v1.ConfigProto(gpu_options=gpu_options))
Run Code Online (Sandbox Code Playgroud)

但这段代码显然不起作用,我正在研究如何使用它。我已经查看了官方张量流网站文档,但它确实令人困惑。有人可以告诉我如何正确使用它吗?

tensorflow

2
推荐指数
1
解决办法
5900
查看次数

标签 统计

tensorflow ×1