无法修复导入错误:无法从Colab上的“albumentations.pytorch”导入名称“ToTensorV2”
我尝试了这个问题的所有三个解决方案来解决问题,但仍然收到错误:
ImportError: cannot import name 'ToTensorV2' from 'albumentations.pytorch' (/usr/local/lib/python3.7/dist-packages/albumentations/pytorch/__init__.py)
我目前的代码:
!pip install -U albumentations
import albumentations
from albumentations.pytorch import ToTensorV2
回答
您是否尝试过出厂重置运行时?如果你还没有,也许 python 仍在加载早期版本。我刚刚在一个新的 colab 实例上测试了链接问题中已接受的解决方案,它运行良好:
!pip install albumentations==0.4.6
import albumentations
from albumentations.pytorch import ToTensorV2
THE END
二维码