Requirements for lerobot

[janchk]

All python packages and their versions

aiohappyeyeballs==2.6.1
aiohttp==3.12.15
aiosignal==1.4.0
annotated-types==0.7.0
astunparse==1.6.3
attrs==25.3.0
av==15.1.0
beniget==0.4.2.post1
blinker==1.9.0
certifi==2025.8.3
chardet==5.2.0
charset-normalizer==3.4.3
click==8.3.0
cloudpickle==3.1.1
cmake==4.1.0
Cython==3.1.3
datasets==3.6.0
deepdiff==8.6.1
diffusers==0.35.1
dill==0.3.8
docopt==0.6.2
draccus==0.10.0
einops==0.8.1
evdev==1.9.2
Farama-Notifications==0.0.4
filelock==3.19.1
Flask==3.1.2
frozenlist==1.7.0
fsspec==2025.3.0
gast==0.6.0
gitdb==4.0.12
GitPython==3.1.45
gymnasium==0.29.1
hf-xet==1.1.10
hf_transfer==0.1.9
huggingface_hub @ file:///buildstream/nutcracker-legacy/deps/python3-huggingface_hub.bst
idna==3.10
imageio==2.37.0
imageio-ffmpeg==0.6.0
importlib_metadata==8.7.0
inquirerpy==0.3.4
itsdangerous==2.2.0
Jinja2==3.1.6
jsonlines==4.0.0
lerobot==0.3.3
Mako==1.3.10.dev0
Markdown==3.9
MarkupSafe==3.0.2
mergedeep==1.3.4
meson==1.9.0
mpmath @ file:///buildstream/nutcracker-legacy/deps/python3-mpmath.bst
multidict==6.6.4
multiprocess==0.70.16
mypy_extensions==1.1.0
networkx==3.5
num2words==0.5.14
numpy==2.2.6
nvidia-cublas-cu12==12.6.4.1
nvidia-cuda-cupti-cu12==12.6.80
nvidia-cuda-nvrtc-cu12==12.6.77
nvidia-cuda-runtime-cu12==12.6.77
nvidia-cudnn-cu12==9.5.1.17
nvidia-cufft-cu12==11.3.0.4
nvidia-cufile-cu12==1.11.1.6
nvidia-curand-cu12==10.3.7.77
nvidia-cusolver-cu12==11.7.1.2
nvidia-cusparse-cu12==12.5.4.2
nvidia-cusparselt-cu12==0.6.3
nvidia-nccl-cu12==2.26.2
nvidia-nvjitlink-cu12==12.6.85
nvidia-nvtx-cu12==12.6.77
onnx==1.16.2
onnxruntime==1.19.0
opencv-python-headless==4.12.0.88
orderly-set==5.5.0
packaging==25.0
pandas==2.3.2
pfzy==0.3.4
pillow==12.0.0.dev0
platformdirs==4.4.0
ply==3.11
prompt_toolkit==3.0.52
propcache==0.3.2
protobuf==6.32.0
psutil==7.1.0
pyarrow==21.0.0
pybind11==3.0.2a0
pydantic==2.11.9
pydantic_core==2.33.2
pynput==1.8.1
pyserial==3.5
python-dateutil==2.9.0.post0
python-xlib==0.33
pythran==0.18.0
pytz==2025.2
PyYAML==6.0.2
pyyaml-include==1.4.1
regex==2025.9.18
requests==2.32.5
rerun-sdk==0.22.1
safetensors==0.6.2
scipy==1.16.3.dev0
sentry-sdk==2.39.0
setuptools==80.9.0
six==1.17.0
smmap==5.0.2
sympy @ file:///buildstream/nutcracker-legacy/deps/python3-sympy.bst
termcolor==3.1.0
tokenizers==0.22.1
toml==0.10.2
torch==2.7.1
torchcodec==0.5
torchvision==0.22.1
tqdm @ file:///buildstream/nutcracker-legacy/deps/python3-huggingface_hub.bst/.bst_pip_downloads/tqdm-4.67.1.tar.gz#sha256=f8aef9c52c08c13a65f30ea34f4e5aac3fd1a34959879d7e59e63027286627f2
transformers @ file:///buildstream/nutcracker-legacy/deps/python3-transformers.bst
triton==3.3.1
typing-inspect==0.9.0
typing-inspection==0.4.1
typing_extensions==4.15.0
tzdata==2025.2
urllib3==2.5.0
wandb==0.22.0
wcwidth==0.2.14
Werkzeug==3.1.3
wheel==0.46.1
xxhash==3.5.0
yarl==1.20.1
zipp==3.23.0

The problem is that lerobot, when installing requires a ton of packages including torch and nvidia-cuda ones..
And if installed package version is differ, they get replaced by the package from pip.

Detailed list of requirements provided by lerobot is here
https://github.com/huggingface/lerobot/blob/b883328e6c95681ca90a18b102e4ae5e1f91e2bf/pyproject.toml#L59-L90

[janchk]

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.

Yes, it's error, but seems that nothing so serious..

[jerenkrantz]

Thanks for this list!

I think that this dependency graph is probably large enough where we'd want to spin a distinct container image with these Python modules layered in on top of the base image. (Perhaps a nutcracker-lerobot or something akin to that.)

We can add this to our backlog - in the interim as long as onnxruntime doesn't get swapped out, a 'pip install' would likely do the right-ish things for now.