当前位置:网站首页>tensorflow 1.14指定gpu运行设置

tensorflow 1.14指定gpu运行设置

2022-07-06 17:10:00 愚昧之山绝望之谷开悟之坡

注意:allow_soft_placement=True表明:计算设备可自行选择,如果没有这个参数,会报错。
因为不是所有的操作都可以被放在GPU上,如果强行将无法放在GPU上的操作指定到GPU上,将会报错。

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"    # 这里是gpu的序号,指定使用的gpu对象
gpu_options = tf.GPUOptions()
gpu_options.allow_growth = True
config_photo = tf.ConfigProto(allow_soft_placement=True, gpu_options=gpu_options, log_device_placement=True)
self.sess = tf.Session(config=config_photo)
原网站

版权声明
本文为[愚昧之山绝望之谷开悟之坡]所创,转载请带上原文链接,感谢
https://blog.csdn.net/qq_15821487/article/details/125637526