diff --git a/tensorflow2fluid/README.md b/tensorflow2fluid/README.md
index 971e189d22ea259b365788f382d2f1acf1c59238..03aa9e276f21c70bec6e227dce0d197bc9aaa383 100644
--- a/tensorflow2fluid/README.md
+++ b/tensorflow2fluid/README.md
@@ -29,6 +29,7 @@ python src/convert.py --pb_file tf_model.pb \
                       --output_nodes outputs \
                       --input_shape None,224,224,3 \
                       --input_format NHWC \
+                      --use_cuda True \
                       --save_dir translated_paddle_model
 ```
 ### 加载模型并预测