[BUG]warn(f“Failed to load image Python extension: {e}“)的解决办法

韬小志2024-10-27 15:32

在使用LlaMa-Factory工具包时，安装好环境后，输入llamafactory-cli env查看llama-factory的版本等信息时，bash提醒：

/home/ubuntu/anaconda3/envs/Llama-Factory/lib/python3.10/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: /home/ubuntu/anaconda3/envs/Llama-Factory/lib/python3.10/site-packages/torchvision/image.so: undefined symbol: _ZN3c1017RegisterOperatorsD1Ev
warn(f"Failed to load image Python extension: {e}")

通过以下命令查看torchvision的版本为0.12.0。

复制代码

pip show torchvision

通过一下命令查看torch的版本为2.5.0。

复制代码

import torch
print(torch.__version__)

打开该网址https://github.com/pytorch/vision#installation查询发现版本不对应。安装对应的版本：

复制代码

pip install torchvision==0.20.0

安装成功之后，再次输入：

复制代码

lamafactory-cli env

上述提醒已经消失。