tensorflow / tensorflow/models

How to run model_main.py with multi-GPU?

Open
#6,047 5 comments 0 reactions 3 assignees View on GitHub

@pkulzc is already working on this.

Since Jul 1, 2020.

models:research:odapi type:support
Dominant language
Python
Stars
77.7k
Forks
44.8k
PR merge metrics
No merged PRs in 30d

Description

System information
  • What is the top-level directory of the model you are using:Object Detection API
  • Have I written custom code (as opposed to using a stock example script provided in TensorFlow): No
  • OS Platform and Distribution (e.g., Linux Ubuntu 16.04):windows 10
  • TensorFlow installed from (source or binary):binary
  • TensorFlow version (use command below):1.12
  • Bazel version (if compiling from source):
  • CUDA/cuDNN version:9.0/7.1
  • GPU model and memory:GTX1080TI 11GB
  • Exact command to reproduce:
Describe the problem

how to run model_main.py with multi-GPU?

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.