ImportError：无法从“torch._six”导入名称“PY3”

Question

我正在使用https://github.com/stereolabs/zed-pytorch上的代码测试 ZED 相机。运行最终命令时： python zed_object_detection.py --config-file configs/caffe2/e2e_mask_rcnn_R_50_C4_1x_caffe2.yaml --min-image-size 256

我收到以下错误：

回溯（最近一次调用最后一次）：文件“zed_object_detection.py”，第 6 行，来自预测器导入 COCODemo 文件“/home/fypadmin/Desktop/23Apr_ZED/zed-pytorch/predictor.py”，第 4 行，来自 torchvision 导入转换为 T 文件“/home/fypadmin/anaconda3/envs/pytorch1/lib/python3.8/site-packages/torchvision/ init .py”，第 4 行，来自 torchvision 导入数据集文件“/home/fypadmin/anaconda3/ envs/pytorch1/lib/python3.8/site-packages/torchvision/datasets/ init .py”，第 1 行，来自 .lsun 导入 LSUN，LSUNClass 文件“/home/fypadmin/anaconda3/envs/pytorch1/lib/python3 .8/site-packages/torchvision/datasets/lsun.py”，第 19 行，来自 .utils import verify_str_arg, iterable_to_str 文件“/home/fypadmin/anaconda3/envs/pytorch1/lib/python3.8/site-packages/ torchvision/datasets/utils.py”，第 11 行，来自 torch._six import PY3 ImportError：无法从 'torch._six' 导入名称 'PY3' (/home/fypadmin/anaconda3/envs/pytorch1/lib/python3.8 /site-packages/torch/_six.py)

我是 ML 新手，正在运行 pytorch 1.8.1。期待任何帮助。谢谢

Answer 1

对于这个问题，原因是你的“torchvision”和“pytorch”版本不匹配。因此，您需要将“torchvision”和“pytorch”版本升级到新版本

pip install --upgrade torch torchvision

Answer 2

您需要将您的 torch 版本降级到 1.4.x 才能兼容。鉴于https://github.com/stereolabs/zed-pytorch大部分是在一两年前编写的，我建议最新使用 python 3.7.x 并降级 torch。