First real commit, simxarm env added with torchrl!

This commit is contained in:
Cadene
2024-01-29 12:49:30 +00:00
parent 0396980450
commit 1144819c29
11 changed files with 437 additions and 1 deletions

23
environment.yaml Normal file
View File

@@ -0,0 +1,23 @@
name: lerobot
dependencies:
- python=3.8.16
- pytorch::pytorch=1.13.1
- pytorch::torchvision=0.14.1
- nvidia::cudatoolkit=11.7
- anaconda::pip
- pip:
- cython==0.29.33
- mujoco==2.3.2
- mujoco-py==2.1.2.14
- termcolor
- omegaconf
- gym==0.21.0
- dm-env==1.6
- pandas
- wandb
- moviepy
- imageio
- gdown
# - -e benchmarks/d4rl
# TODO: verify this works
- git+https://github.com/nicklashansen/simxarm.git@main#egg=simxarm