added conda env

2023-04-11 15:00:56 -04:00
parent e3bf8dec54
commit e4f8df66de
1 changed files with 24 additions and 0 deletions
--- a/conda_env.yaml
+++ b/conda_env.yaml
@@ -0,0 +1,24 @@
+name: aloha
+channels:
+  - pytorch
+  - nvidia
+  - conda-forge
+dependencies:
+  - python=3.9
+  - pip=23.0.1
+  - pytorch=2.0.0
+  - torchvision=0.15.0
+  - pytorch-cuda=11.8
+  - pyquaternion=0.9.9
+  - pyyaml=6.0
+  - rospkg=1.5.0
+  - pexpect=4.8.0
+  - mujoco=2.3.3
+  - dm_control=1.0.9
+  - py-opencv=4.7.0
+  - matplotlib=3.7.1
+  - einops=0.6.0
+  - packaging=23.0
+  - h5py=3.8.0
+  - pip:
+    - h5py_cache==1.0