search

NVlabs / trajdata

A unified interface to many trajectory forecasting datasets.
Apache License 2.0
290 stars 38 forks source link

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

Open rudrakhsha-crest opened 1 month ago

rudrakhsha-crest commented 1 month ago

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