Skip to content

Latest commit

 

History

History
95 lines (75 loc) · 2.67 KB

README.md

File metadata and controls

95 lines (75 loc) · 2.67 KB
Raw

Comparative study of Dreamer V3, PlaNet, and PPO in tasks involving object deformation

Writeup: https://drive.google.com/file/d/1EpgLxWMRQ9l8iS19pNttOWgUIW9-mSA9/view?usp=sharing

Installation

  • clone this repo:
https://github.com/SergioMOrozco/decision_making_project.git
cd decision_making_project

Create Environment

  • create environment and install softgym dependencies
conda create -n softgym python=3.11
conda activate softgym

Install PyFleX For Softgym

  • install docker-ce, then:
sudo docker pull xingyu/softgym
  • install pybind11 like so:
conda install pybind11
  • run docker container like so (obviously change to where PyFleX and anacodna is installed on your machine):
sudo docker run -v ~/dev/decision_making_project/softagent/softgym/:/workspace/softgym/ -v ~/anaconda3/envs/softgym/:/workspace/anaconda3/envs/softgym/ -v /tmp/.X11-unix/:/tmp/.X11-unix/ --gpus all -e DISPLAY=$DISPLAY -e QT_X11_NO_MITSHM=1 -it xingyu/softgym:latest bash
  • Compile PyFleX in docker environment:
export PATH="/workspace/anaconda3/envs/softgym/bin:$PATH"
cd softgym
. ./prepare_1.0.sh && ./compile_1.0.sh
  • exit the docker container, and add the following to ~/.bashrc:
export SOFTGYMROOT=/home/sorozco0612/dev/decision_making_project/softagent/softgym
export DREAMERROOT=/home/sorozco0612/dev/decision_making_project/softagent/dreamerv3
export PYFLEXROOT=${SOFTGYMROOT}/PyFlex
export PYTHONPATH=${DREAMERROOT}:${SOFTGYMROOT}:${PYFLEXROOT}/bindings/build:$PYTHONPATH
  • source ~/.basrhc like so:
source ~/.bashrc
conda activate softgym
  • confirm that softgym can run with PyFlex by running the following example:
cd softagent/softgym/
python examples/random_env.py --env_name PassWater
  • keep installing missing dependencies as they come up. Rather than working with environment.yml we let pip and conda automatically resolve dependencies and versions

Install Dreamer Dependencies

  • install JAX for NVIDIA GPU hardware:
conda activate softgym
pip install -U "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
  • install Gym dependencies:
python -m pip install --upgrade pip setuptools wheel

Confirm Installation

  • You can confirm that dreamerV3 was installed correctly by running the following without error:
cd softagent/dreamerv3/
python example.py
  • keep installing missing dependencies as they come up. Rather than working with requirements.txt we let pip and conda automatically resolve dependencies and versions

Running PlaNet

To reproduce our results on PlaNet run the following script

cd softagents
./run_expts.sh

Install dependencies as prompted