From 97d8eeb30c3c0e6e3b336346b67fb4e000178ee5 Mon Sep 17 00:00:00 2001
From: Stephen Tyree <styree@nvidia.com>
Date: Mon, 23 Jan 2023 15:41:12 -0600
Subject: [PATCH] working conda env

---
 environment.yaml | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/environment.yaml b/environment.yaml
index b67b4ed..f6e78c4 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -5,9 +5,8 @@ channels:
 dependencies:
   - python=3.7.6
   - pinocchio=2.3.1
+  - urdfdom=1.0.4
   - assimp=4.1.0
-  - pytorch=1.3.1=py3.7_cuda10.1.243_cudnn7.6.3_0
-  - torchvision=0.4.2=py37_cu101
   - rclone=1.52.3
   - gxx_linux-64=7.3.0
   - git-lfs=2.11.0
@@ -20,8 +19,10 @@ dependencies:
   - pillow=6.0.0
   - ipython
   - pip:
-      - ./deps/bullet3
       - ./deps/job-runner
+      - torch
+      - torchvision
+      - pybullet
       - imageio==2.6.1
       - simplejson==3.17.0
       - opencv-python==4.1.2.30