0
点赞
收藏
分享

微信扫一扫

tensorflow 单机多GPU 实例

高子歌 2022-07-27 阅读 110


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

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

170行左右开始重点


举报

相关推荐

0 条评论