-
Notifications
You must be signed in to change notification settings - Fork 1.4k
RuntimeError: Could not load libtorchcodec during lerobot/scripts/train.py script #964
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Comments
Try with latest main and new installation instructions (: |
I have the exact same issue. But I think it is due to some problem with the torchcodec. |
Hi @gpthimble ! Can you confirm that you installed
Thanks, Caroline. |
I have tried the new instructions in the readme, build a new enviroment, install the ffmpeg 7.1.1. |
I think you should try different version of Torchcodec. version 0.2.1 is for the torch version 2.6 not 2.5. |
Thanks for pointing that out ! We're going to solve all I'm closing the issue as it is solved now. Best, Caroline. |
I have also tried lerobot project on an AMD GPU with rocm, with no luck. I think the torch itself should work without problem, but it seems that the package torchcodec is not compactable with rocm version of torch. I will post an issue on the torchcodec repository about that. Really hope this can work on an amd gpu! |
System Info
Information
Reproduction
Install leRobot from the main documentation as follows :
conda create -n lerobot python=3.10 -y
conda activate lerobot
git clone https://github.com/huggingface/lerobot.git ~/lerobot
pip install --no-binary=av -e
pip install torchvision==0.20.1
conda install -c conda-forge 'ffmpeg>=7.0' -y
After collecting a dataset, run
lerobot/scripts/train.py
scriptExpected behavior
Hello all!
I am getting started with the lerobot so100 arm and have had a few issues.
The first was the same as the issue in #883 in running the
control_robot.py
script which I solved (or bypassed) by following remi cadene's response to dopip install torchvision==0.20.1
and alsoconda install -c conda-forge 'ffmpeg>=7.0' -y
after doingpip install --no-binary=av -e
. This allowed me to successfully run thecontrol_robot.py
script successfully. However, then I tried to collect a dataset and run a training with thelerobot/scripts/train.py
script and I encountered the following issue :It seems that I have some issues with the
torchcodec
andffmpeg
versions not being compatible. Checking their versions gives me:And
TorchCodec
version 0.2.1.Could anyone suggest the right versions to install and whether I should downgrade ffmpeg ?
The text was updated successfully, but these errors were encountered: