解决 'torchvision' 加载图像功能失败警告：缺少 'libjpeg' 或 'libpng' 库

在使用 'torchvision' 时，可能会遇到以下警告：

'UserWarning: Failed to load image Python extension: '/home/ubuntu/miniconda3/envs/py39w/lib/python3.9/site-packages/torchvision/image.so: undefined symbol: _ZN3c104warnERKNS_7WarningE' If you don't plan on using image functionality from 'torchvision.io', you can ignore this warning. Otherwise, there might be something wrong with your environment. Did you have 'libjpeg' or 'libpng' installed before building 'torchvision' from source? warn(

这个警告表明 'torchvision' 尝试加载一个 Python 扩展的图像功能，但是失败了。原因是当前环境缺少 'libjpeg' 或 'libpng' 库。这可能是由于以下原因导致的：

• 在构建 'torchvision' 时没有安装 'libjpeg' 或 'libpng' 库。
• 'libjpeg' 或 'libpng' 库没有正确地配置到当前环境中。

要解决这个问题，可以采取以下步骤：

1. 安装 'libjpeg' 和 'libpng' 库：

   • 使用系统包管理器安装：例如，在 Ubuntu 中可以使用 sudo apt-get install libjpeg-dev libpng-dev 命令进行安装。
   • 从源代码编译安装：如果系统包管理器没有提供所需的库版本，则可以从源代码编译安装。

2. 重新构建 'torchvision'：

   • 在安装 'libjpeg' 和 'libpng' 库后，需要重新构建 'torchvision'。这可以通过以下步骤完成：
     • 卸载现有的 'torchvision' 包：pip uninstall torchvision
     • 重新安装 'torchvision'：pip install torchvision

3. 检查环境配置：

   • 确保 'libjpeg' 和 'libpng' 库的路径被添加到环境变量中。
   • 检查系统是否正确识别了这些库。

完成以上步骤后，重新运行代码，应该可以解决 'Failed to load image Python extension' 警告。

如果问题仍然存在，可以尝试查看 'torchvision' 的文档或寻求更专业的技术支持。