pip - 无法修复导入错误 : cannot import name 'ToTensorV2' from 'albumentations.pytorch' on Colab

Question

我尝试了来自 this question 的所有三种解决方案解决了问题，但仍然收到错误:

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

Answer 1

您是否尝试过出厂重置运行时？如果你还没有，也许 python 仍在加载早期版本。我刚刚在一个新的 colab 实例上测试了链接问题中接受的解决方案，它运行良好:

!pip install albumentations==0.4.6
import albumentations 
from albumentations.pytorch import ToTensorV2