zoukankan      html  css  js  c++  java
  • tensorflow 单机多卡 官方cifar10例程

    测试了官方历程,看没有问题,加上时间紧任务重,就不深究了。

    官方tutorials:https://www.tensorflow.org/tutorials/images/deep_cnn

    github源码:https://github.com/tensorflow/models/tree/master/tutorials/image/cifar10

    值得注意的是:

    • 运行多GPU训练的脚本:
      python cifar10_multi_gpu_train.py --num_gpus = 1 --batch_size=4096
    • 在有4个1080TI的服务器上 运行如下shell命令,后者速度明显提升。
      python cifar10_multi_gpu_train.py --num_gpus=1 --max_steps=100  --batch_size=4096
      python cifar10_multi_gpu_train.py --num_gpus=2 --max_steps=100  --batch_size=2048
      python cifar10_multi_gpu_train.py --num_gpus=3 --max_steps=100  --batch_size=1024
      python cifar10_multi_gpu_train.py --num_gpus=4 --max_steps=100  --batch_size=512

      结果为:

      num_gpus batch_size min msec/batch
      1 4096 256
      2 2048 64
      3 1024 29
      4 512 15
  • 相关阅读:
    Redis学习小结
    抽屉模型
    用户提交数据的验证
    jsonp原理与实验
    文件上传
    项目
    CBV
    C++算法 线段树
    写一些奇怪的东西找到的奇怪的错误
    php安装过程出现的一些错误问题:
  • 原文地址:https://www.cnblogs.com/xbit/p/10107877.html
Copyright © 2011-2022 走看看