From e3137b63478cdf882a9e1f02967312332b145c01 Mon Sep 17 00:00:00 2001
From: tink2123 <y_tink@163.com>
Date: Mon, 22 Apr 2019 15:44:29 +0800
Subject: [PATCH] modified readme

---
 PaddleCV/yolov3/README.md    | 2 ++
 PaddleCV/yolov3/README_cn.md | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/PaddleCV/yolov3/README.md b/PaddleCV/yolov3/README.md
index 7bf917e9..109c8dba 100644
--- a/PaddleCV/yolov3/README.md
+++ b/PaddleCV/yolov3/README.md
@@ -155,6 +155,8 @@ Inference is used to get prediction score or image features based on trained mod
         --image_name=000000000139.jpg \
         --draw_threshold=0.5
 
+- Set ```export CUDA_VISIBLE_DEVICES=0``` to specifiy one GPU to infer.
+
 Inference speed（Tesla P40）:
 
 
diff --git a/PaddleCV/yolov3/README_cn.md b/PaddleCV/yolov3/README_cn.md
index 4cbbf7f1..cfb01ee9 100644
--- a/PaddleCV/yolov3/README_cn.md
+++ b/PaddleCV/yolov3/README_cn.md
@@ -157,6 +157,8 @@ Train Loss
         --image_name=000000000139.jpg \
         --draw_threshold=0.5
 
+- 通过设置export CUDA\_VISIBLE\_DEVICES=0指定单卡GPU预测。
+
 模型预测速度（Tesla P40）：
 
 
-- 
GitLab