logo
  • 教程列表
  • SO官方文档
  • TensorFlow GPU 设置
    • 控制 GPU 内存分配
    • 仅在 CPU 上运行 TensorFlow - 使用 CUDA VISIBLE DEVICES 环境变量
    • 仅在 CPU 上运行 TensorFlow Graph - 使用 tf.config
    • 使用一组特定的 GPU 设备
    • 列出 TensorFlow 在本地流程中可用的设备
  1. StackOverflow 文档
  2. tensorflow 教程
  3. TensorFlow GPU 设置

TensorFlow GPU 设置

Created: November-22, 2018

主要来源:

  • https://www.tensorflow.org
  • https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/protobuf/config.proto
  • https://stackoverflow.com/a/37901914
  • https://github.com/tensorflow/tensorflow/issues/152
  • https://github.com/tensorflow/tensorflow/issue/9201
  • 控制 GPU 内存分配

Copyright © 2018. All right reserved

tastones.com 备案号:鲁ICP备18045372号-1

  • 关于我们
  • 免责声明