[BUG]warn(f“Failed to load image Python extension: {e}“)的解决办法
在使用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
上述提醒已经消失。