NVlabs / trajdata

A unified interface to many trajectory forecasting datasets.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Time taken to install trajdata[waymo] is too long.

rudrakhsha-crest opened this issue · comments

Hi everyone
I am trying to install the devkit for waymo motion dataset using pip install "trajdata[waymo]" but pip is downloading each version of the requirements specified in the pyproject.toml file instead of the available ones. This is taking way too long and even loading nvidia packages from cache is taking north of 10-20 mins per package from the cache.
Here are the dependancies I am trying to install from the .toml file.

numpy>=1.19
tqdm>=4.62
matplotlib>=3.5
dill>=0.3.4
pandas>=1.4.1
pyarrow>=7.0.0
torch>=1.10.2
zarr>=2.11.0
kornia>=0.6.4
seaborn>=0.12
bokeh>=3.0.3
geopandas>=0.13.2
protobuf==3.19.4
scipy>=1.9.0
opencv-python>=4.5.0
shapely>=2.0.0

Can someone help me with this issue? Is there anything I might be doing incorrectly while installing the waymo package?
Any help would be greatly appretiated. Thanks