diff --git a/docker/scripts/install_demo.sh b/docker/scripts/install_demo.sh index 5110e910..809a24f8 100644 --- a/docker/scripts/install_demo.sh +++ b/docker/scripts/install_demo.sh @@ -13,9 +13,10 @@ pip install --no-cache-dir --upgrade numpy==1.23.0 export TMPDIR='/var/tmp' # Pip install VGN requirements -pip install --no-cache-dir catkin_pkg scipy pandas matplotlib mpi4py open3d pybullet==2.7.9 tqdm +pip install --no-cache-dir catkin_pkg scipy pandas==0.22.0 matplotlib mpi4py open3d pybullet==2.7.9 tqdm pip cache purge -pip install --no-cache-dir torch pytorch-ignite tensorboard +pip install --no-cache-dir torch==1.2.0+cpu -f https://download.pytorch.org/whl/torch_stable.html +pip install tensorboard #pytorch-ignite # Pip install gdown to get google drive files pip install gdown