diff --git a/README.md b/README.md
index bc20c1ce..c31196e9 100644
--- a/README.md
+++ b/README.md
@@ -250,7 +250,7 @@ export USE_MODELSCOPE_HUB=1 # `set USE_MODELSCOPE_HUB=1` for Windows
 Then you can train the corresponding model by specifying a model ID of the ModelScope Hub. (find a full list of model IDs at [ModelScope Hub](https://modelscope.cn/models))
 
 ```bash
-python src/train_bash.py \
+CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
     --model_name_or_path modelscope/Llama-2-7b-ms \
     ... # arguments (same as above)
 ```
diff --git a/README_zh.md b/README_zh.md
index 0aa8ac0f..6cc7a628 100644
--- a/README_zh.md
+++ b/README_zh.md
@@ -250,7 +250,7 @@ export USE_MODELSCOPE_HUB=1 # Windows 使用 `set USE_MODELSCOPE_HUB=1`
 接着即可通过指定模型名称来训练对应的模型。（在[魔搭社区](https://modelscope.cn/models)查看所有可用的模型）
 
 ```bash
-python src/train_bash.py \
+CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
     --model_name_or_path modelscope/Llama-2-7b-ms \
     ... # 参数同上
 ```