First real commit, simxarm env added with torchrl!

2024-01-29 12:49:30 +00:00
parent 0396980450
commit 1144819c29
11 changed files with 437 additions and 1 deletions
--- a/environment.yaml
+++ b/environment.yaml
@@ -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