diff --git a/docs/en/PyTorch Installation Guide/PyTorch Installation Guide.md b/docs/en/PyTorch Installation Guide/PyTorch Installation Guide.md
index 1dbf866d8c41e8c8d0c972572bdc845f4a6b455c..85f8a4402c556629110bf8a837919a953c95a3c7 100644
--- a/docs/en/PyTorch Installation Guide/PyTorch Installation Guide.md	
+++ b/docs/en/PyTorch Installation Guide/PyTorch Installation Guide.md	
@@ -150,7 +150,7 @@ When setting up the environment for PyTorch model development and running, you c
         bash build.sh --python=3.8
         or
         bash build.sh --python=3.9    # PyTorch 1.5 does not support build and installation using Python 3.9.
-
+        
         ```
 
         Specify the Python version in the environment for build. The generated binary package is stored in the current dist directory  **pytorch/pytorch/dist**.
@@ -339,7 +339,7 @@ After the software packages are installed, configure environment variables to us
     1.  Run the following command to obtain the Apex source code adapted to Ascend AI Processors:
 
         ```
-        git clone https://gitee.com/ascend/apex.git
+        git clone -b 2.0.4.tr5 https://gitee.com/ascend/apex.git
         ```
 
         The directory structure of the downloaded source code is as follows:
diff --git "a/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md" "b/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md"
index 23432e483a70006a10f481e1440881244758c356..9cbd60a7cbb25f46a8b755f8467e4ed58149b9e0 100644
--- "a/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md"
+++ "b/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md"
@@ -352,7 +352,7 @@
     1.  运行如下命令，获取适配昇腾AI处理器的apex源代码。
 
         ```
-        git clone https://gitee.com/ascend/apex.git
+        git clone -b 2.0.4.tr5 https://gitee.com/ascend/apex.git
         ```
 
         下载的源码主要目录结构如下所示：