torch version required to 1.10

1292224662 · Apr 9, 2023 · 8133e30 · 8133e30
1 parent a7c83ea
commit 8133e30
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/0.get_started.md b/docs/0.get_started.md
@@ -5,7 +5,7 @@
     git clone https://github.com/ShiqiYu/OpenGait.git
     ```
 2. Install dependenices:
-    - pytorch >= 1.6
+    - pytorch >= 1.10
     - torchvision
     - pyyaml
     - tensorboard
@@ -17,12 +17,12 @@
     Install dependenices by [Anaconda](https://conda.io/projects/conda/en/latest/user-guide/install/index.html):
     ```
     conda install tqdm pyyaml tensorboard opencv kornia -c conda-forge
-    conda install pytorch==1.6.0 torchvision -c pytorch
+    conda install pytorch==1.10 torchvision -c pytorch
     ```    
     Or, Install dependenices by pip:
     ```
     pip install tqdm pyyaml tensorboard opencv-python kornia
-    pip install torch==1.6.0 torchvision==0.7.0
+    pip install torch==1.10 torchvision==0.11
     ```
 ## Prepare dataset
 See [prepare dataset](2.prepare_dataset.md).