From e4f8df66de6f59b9c6908891472c549af2ec14da Mon Sep 17 00:00:00 2001 From: Cheng Chi Date: Tue, 11 Apr 2023 15:00:56 -0400 Subject: [PATCH] added conda env --- conda_env.yaml | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 conda_env.yaml diff --git a/conda_env.yaml b/conda_env.yaml new file mode 100644 index 0000000..9ab88c8 --- /dev/null +++ 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