torch version required to 1.10

This commit is contained in:
darkliang
2023-04-09 18:57:41 +08:00
parent a7c83ea7f4
commit 8133e3017f
+3 -3
View File
@@ -5,7 +5,7 @@
git clone https://github.com/ShiqiYu/OpenGait.git git clone https://github.com/ShiqiYu/OpenGait.git
``` ```
2. Install dependenices: 2. Install dependenices:
- pytorch >= 1.6 - pytorch >= 1.10
- torchvision - torchvision
- pyyaml - pyyaml
- tensorboard - tensorboard
@@ -17,12 +17,12 @@
Install dependenices by [Anaconda](https://conda.io/projects/conda/en/latest/user-guide/install/index.html): 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 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: Or, Install dependenices by pip:
``` ```
pip install tqdm pyyaml tensorboard opencv-python kornia 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 ## Prepare dataset
See [prepare dataset](2.prepare_dataset.md). See [prepare dataset](2.prepare_dataset.md).