From 97ab3dae831771d6427aa9acc3e0b2ae4205a2b1 Mon Sep 17 00:00:00 2001
From: chuboning <chuboning@foxmail.com>
Date: Mon, 20 Nov 2023 17:40:55 +0800
Subject: [PATCH] Add torch version config in readme

---
 scripts/docker/README.md | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/scripts/docker/README.md b/scripts/docker/README.md
index 6330ae8..d6da4ae 100644
--- a/scripts/docker/README.md
+++ b/scripts/docker/README.md
@@ -20,10 +20,11 @@ docker run -it -v /{code_path}/apex:/home/apex manylinux-builder:v1 bash
 ```
 **Install torch**
 
-Take Python 3.8 as an example
+Take Python 3.8 and torch 2.1.0 as an example
 ```Shell
-pip3.8 install torch
+pip3.8 install torch==2.1.0
 # Reference url: https://pytorch.org/get-started/previous-versions/
+# Please modify the installation commands based on the torch version
 ```
 **Compile apex**
 ```Shell
-- 
Gitee