CompressAI 报错 UserWarning: Failed to load image Python extension: [WinError 127]

问题描述

在运行compressAI的codec.py --help时报错

UserWarning: Failed to load image Python extension: [WinError 127] 找不到指定的程序。

---

原因分析：

查阅得知torch和torchvision版本不对应导致问题,版本对应关系如下

---

解决方案：

按照表中对应关系更新torchvision版本即可

pip

pip uninstall torchvision
pip install torchvision=='版本号' -i https://pypi.tuna.tsinghua.edu.cn/simple/

conda

conda uninstall torchvison
conda install torchvison=='版本号'

参考博客

[1] https://blog.csdn.net/jorg_zhao/article/details/106883420