Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,3 +1,6 @@
# Custom
diffusion_policy

# Logging
logs
tmp
Expand Down
44 changes: 12 additions & 32 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,52 +2,32 @@

## Installation

Install dependencies using `conda`:

Create a virtual environment with python 3.10, e.g. using `conda`:
```
conda env create -f environment.yaml
conda create -y -n lerobot python=3.10
conda activate lerobot
```

Install `torchrl`, `tensordict` and `diffusion_policy` dev builds
[Install `poetry`](https://python-poetry.org/docs/#installation) (if you don't have it already)
```
cd path/to/root
git clone https://github.com/pytorch/tensordict
git clone https://github.com/pytorch/rl
git clone https://github.com/real-stanford/diffusion_policy
cd tensordict
python setup.py develop
cd ../rl
python setup.py develop
cd ../diffusion_policy
python setup.py develop
curl -sSL https://install.python-poetry.org | python3 -
```

Install additional modules
Install dependencies
```
pip install \
hydra \
termcolor \
einops \
pygame \
pymunk \
zarr \
gym \
shapely \
opencv-python \
scikit-image \
mpmath==1.3.0 \
poetry install
```

Fix Hydra
If you encounter a disk space error, try to change your tmp dir to a location where you have enough disk space, e.g.
```
pip install hydra-core --upgrade
mkdir ~/tmp
export TMPDIR='~/tmp'
```

**dev**

Install `diffusion_policy` #HACK
```
python setup.py develop
git clone https://github.com/real-stanford/diffusion_policy
cp -r diffusion_policy/diffusion_policy $(poetry env info -p)/lib/python3.10/site-packages/
```

## Usage
Expand Down
Loading