Tensorflow 可否使用共享 GPU 内存？

技术问答
1

平时训练的时候，Tensorflow 均只使用了专用 GPU 内存，是否有办法使用共享 GPU 内存？

winter， 2018-6-7 23:00:04

2

什么叫共享的 gpu 内存呀。allow growth 选项允许 tf 动态规划内存使用。对您有帮助吗。

winter: 打开任务管理器，GPU 页面，有一个共享 GPU 内存。
2018-6-10 19:07

舟 3332 发表于 2018-6-8 21:22:27

3

可以看下：

使用则是通过 ConfigProto.

gpu_options=tf.GPUOptions (per_process_gpu_memory_fraction=0.7)
config=tf.ConfigProto (gpu_options=gpu_options)
session = tf.Session (config=config, …)

winter: 谢谢，我去看一下
2018-6-10 19:08

HeyJobs: 回复 winter :大兄弟，验证成功否？
2019-9-2 22:25

dcmrlee 发表于 2018-6-9 15:11:55

4

你是说用 GPU 计算吧，GPU 有算力，CPU 只能操作北桥挂的内存。

neverchange 发表于 2018-7-3 18:24:03

5

答案是可以的。。。你要是没写多 gpu 运算，然后你的程序一定要跑在多个 gpu，那一定是共享了显存

ViolinSolo 发表于 2018-7-3 20:29:17