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
```
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).