60 lines
1.5 KiB
Markdown
60 lines
1.5 KiB
Markdown
# Installation
|
||
|
||
## Prerequisites & Installation
|
||
|
||
```bash
|
||
conda create -n ezv2 python=3.8
|
||
conda activate ezv2
|
||
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2
|
||
pip install -r requirements.txt
|
||
```
|
||
|
||
Before starting training, you need to build the c++/cython style external packages.
|
||
```
|
||
cd ez/mcts/ctree
|
||
bash make.sh
|
||
cd -
|
||
```
|
||
|
||
|
||
## Some Tips
|
||
|
||
1. Install `gym` version 0.22.0 (previously 0.15.7, which caused the error: `cannot import name ‘Monitor’ from 'gym.wrappers'`):
|
||
```bash
|
||
pip install gym==0.22.0
|
||
```
|
||
|
||
2. Install `wandb` version 0.14.2 (the previous environment had no version restrictions, but needs downgrading to avoid `TypeError`):
|
||
```bash
|
||
pip install wandb==0.14.2
|
||
```
|
||
|
||
3. Install `reference` version 0.31.0 (version 0.30.2 fails to import `referencing.jsonschema`):
|
||
```bash
|
||
pip install reference==0.31.0
|
||
```
|
||
|
||
4. Install `dmc2gym` from GitHub:
|
||
```bash
|
||
pip install git+git://github.com/denisyarats/dmc2gym.git
|
||
```
|
||
|
||
5. Install `imageio` with `ffmpeg` and `pyav` to ensure `eval_worker` runs correctly:
|
||
```bash
|
||
pip install imageio[ffmpeg]
|
||
pip install imageio[pyav]
|
||
```
|
||
|
||
## Compilation Instructions
|
||
|
||
1. In addition to compiling `ctree`, also compile `ctree_v2` (If you find the training process is stuck, this issue exists in some cases.):
|
||
```bash
|
||
cd ez/mcts/ctree_v2
|
||
sh make.sh
|
||
```
|
||
1. In addition to compiling `ctree`, also compile `ori_ctree` :
|
||
```bash
|
||
cd ez/mcts/ori_ctree
|
||
sh make.sh
|
||
```
|