ImportError: cannot import name 'VGG19_Weights' from 'torchvision.models'

Question

ImportError: cannot import name 'VGG19_Weights' from 'torchvision.models' (C:\Users\86176.conda\envs\pytorch\lib\site-packages\torchvision\models_init_.py)

Answer 1

m0_61940189 下午好🌅🌅🌅
本答案参考ChatGPT-3.5

这个错误通常发生在使用torchvision库中的models模块导入VGG19_Weights时。根据错误提示，找不到名为'VGG19_Weights'的导入模块。

解决这个问题的步骤如下：

1. 首先，确保你安装了正确版本的torchvision和torch库。你可以通过在命令行运行以下命令来检查当前安装的版本：

   pip show torchvision
   pip show torch
   

   如果版本不正确或者你还没有安装这些包，你可以使用以下命令升级或安装：

   pip install torchvision --upgrade
   pip install torch --upgrade
   

2. 确认你在导入VGG19模型权重时使用的是正确的导入名称。根据错误提示，你可能需要将'VGG19_Weights'更改为正确的导入名称。

   在torchvision库中，VGG19模型的权重被命名为'vgg19'，因此正确的导入名称应该是'vgg19'而不是'VGG19_Weights'。

3. 如果你已确认使用了正确的导入名称但仍然出现错误，请检查torchvision包的版本兼容性问题。有时候不同版本的torch库与torchvision库不兼容，导致无法正确导入模型权重。

   你可以尝试通过以下步骤解决这个问题：

   • 升级torchvision库到最新版本：

     pip install torchvision --upgrade
     

   • 指定torchvision库的版本来安装：

     pip install torchvision==0.8.1
     

   • 升级torch库到torchvision库所需的兼容版本：

     pip install torch --upgrade
     

   • 回退torch库的版本到兼容的版本：

     pip install torch==1.7.1
     

   请根据你的情况选择上述步骤中的一种方法，以解决版本兼容性问题。

4. 如果以上解决方法仍然无法解决问题，请确认你已正确安装并配置了CUDA环境。在某些情况下，缺少或不正确配置的CUDA环境可能导致无法导入模型权重。

   解决方法如下:

   • 确保你已正确安装了支持你GPU的NVIDIA驱动程序。
   • 确保你已正确安装了与你的CUDA版本相匹配的CUDA工具包。
   • 确保你的CUDA环境变量已正确设置。

   请根据你的情况检查和调整CUDA环境配置。

以上是解决"ImportError: cannot import name 'VGG19_Weights' from 'torchvision.models'"的一些常见方法。根据错误提示和你的实际情况，你可以尝试上述方法来解决这个问题。