问题

我正在学习train an object detection model for Lens Studio的教程，我想使用GPU来训练模型，可以是Google Colab的云GPU，也可以是Jupyter的本地GPU。
尝试导入时：

import torchvision
from torchvision.models.mobilenet import ConvBNReLU
from torchvision.ops import box_iou, nms

当运行时设置为GPU时，我得到了以下错误。

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-3-157e58f8c700> in <module>
     26 
     27 import torchvision
---> 28 from torchvision.models.mobilenet import ConvBNReLU
     29 from torchvision.ops import box_iou, nms
     30 

ImportError: cannot import name 'ConvBNReLU' from 'torchvision.models.mobilenet' (/usr/local/lib/python3.8/dist-packages/torchvision/models/mobilenet.py)

---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------

预期行为：

它应该只是返回进口。

我尝试了什么：

我尝试使用Jupyter和我的本地GPU在本地环境中运行代码，并通过使用Jupyter将本地运行时连接到Google Colab以使用本地GPU。在这两种情况下，导入错误仍然存在。
PyTorch在本地环境中的设置确认CUDA和本地GPU的使用已启用并可用。
当我将运行时设置为CPU时，所有内容都被导入并按预期工作，我可以训练和导出模型。
因此，我认为这个问题与GPU的使用和

import 'ConvBNReLU' from 'torchvision.models.mobilenet’.

我将非常感谢你的任何建议。