平时训练的时候,Tensorflow均只使用了专用GPU内存,是否有办法使用共享GPU内存?
提问人:winter
发帖时间:2018-6-7 23:00:04
平时训练的时候,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