ludwig
ludwig copied to clipboard
The torchtext version 0.14.1 in the Docker image is incompatible with pytorch 2.0.0
the command to reproduce:
podman run --entrypoint python ludwig-gpu:release-0.7 -c "import torchtext; print(torchtext.__version__);"
Output:
Traceback (most recent call last):
File "
fast workaround:
exec pip install --force-reinstall --no-deps torchtext==0.15.1
in the container
Thanks @abaletinskikh for posting the issue and for posting a workaround.
Can confirm that version compatibility is particularly important for torch libraries, and indeed the latest compatible version of torchtext that we support is v0.15.1.
I notice that you are pulling down the python ludwig-gpu:release-0.7
docker image. Was there something about the latest docker image ludwigai/ludwig-gpu:master
that didn't work for you?
EDIT:
Actually, there have been several dependency updates since the last 0.7 release. I'll recommend continuing to use the ludwigai/ludwig-gpu:release-0.7
as the ludwigai/ludwig-gpu:master
image is under active development.
Hey @abaletinskikh, to clarify something, ludwig-gpu:release-0.7
is a bit of an older image, I would recommend updating to ludwig-gpu:release-0.7.4
, which should be the latest official image. Can you try using this image and letting us know if you see the same issue?