EfficientZeroV2/INSTALL.md

60 lines
1.5 KiB
Markdown
Raw Normal View History

2024-06-07 16:02:01 +08:00
# 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
```
2024-08-09 21:55:45 +08:00
1. In addition to compiling `ctree`, also compile `ori_ctree` :
```bash
cd ez/mcts/ori_ctree
sh make.sh
```