Tensorflow可否使用共享GPU内存?

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

提问人:winter
发帖时间:2018-6-7 23:00:04

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


回复人:舟3332
发帖时间:2018-6-8 21:22

打开任务管理器,GPU页面,有一个共享GPU内存。


发帖人:winter
发帖时间:2018-6-10 19:07

可以看下: config.proto

使用则是通过 ConfigProto.

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

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

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


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

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


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