tensorflow 单机多GPU 实例

来源:互联网 发布:域名对应ip查询 编辑:程序博客网 时间:2024/06/06 09:44

tensorflow 默认 单机 占用多个GPU的内存,实际只一个GPU在跑,要自己改程序并行化

https://github.com/tensorflow/models/blob/master/tutorials/image/cifar10/cifar10_multi_gpu_train.py

170行左右开始重点

原创粉丝点击