发新帖

Tensorflow可否使用共享GPU内存?

[复制链接]
418 4

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
平时训练的时候,Tensorflow均只使用了专用GPU内存,是否有办法使用共享GPU内存?
我知道答案 回答被采纳将会获得10 金币 + 32 金币 已有4人回答
本楼点评(0) 收起

精彩评论4

舟3332  TF芽芽  发表于 2018-6-8 21:22:27 来自手机  | 显示全部楼层
什么叫共享的 gpu 内存呀。allow growth 选项允许 tf 动态规划内存使用。对您有帮助吗。
本楼点评(1) 收起
dcmrlee  TF荚荚  发表于 2018-6-9 15:11:55 | 显示全部楼层
可以看下: https://github.com/tensorflow/te ... uf/config.proto#L21
使用则是通过 ConfigProto.

gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config=tf.ConfigProto(gpu_options=gpu_options)
session = tf.Session(config=config, ...)
本楼点评(1) 收起
neverchange  TF豆豆  发表于 2018-7-3 18:24:03 | 显示全部楼层
你是说用GPU计算吧,GPU有算力,CPU只能操作北桥挂的内存。
本楼点评(0) 收起
ViolinSolo  TF豆豆  发表于 2018-7-3 20:29:17 | 显示全部楼层
答案是可以的。。。你要是没写多gpu运算,然后你的程序一定要跑在多个gpu,那一定是共享了显存
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

36

积分
快速回复 返回顶部 返回列表