python - 升级到 PyTorch 1.9 ImportError 时的潜在错误:无法从“torch._six”导入名称“int_classes”
问题描述
我使用安装 PyTorch
$ pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
I get this error:
(proxy) [jalal@goku proxynca_pp]$ CUDA_VISIBLE_DEVICES=0,1 python train.py --dataset cub --config config/cub.json --mode train --apex --seed 0
Traceback (most recent call last):
File "train.py", line 3, in <module>
import dataset
File "/scratch3/research/code/fashion/proxynca_pp/dataset/__init__.py", line 6, in <module>
from . import utils
File "/scratch3/research/code/fashion/proxynca_pp/dataset/utils.py", line 8, in <module>
from torch._six import int_classes as _int_classes
ImportError: cannot import name 'int_classes' from 'torch._six' (/scratch3/venv/proxy/lib/python3.8/site-packages/torch/_six.py)
代码来自此 GitHub 存储库。我应该更改代码中的哪些内容才能使其正常工作?
(proxy) [jalal@goku proxynca_pp]$ pip freeze
h5py==3.4.0
numpy==1.21.2
Pillow==8.3.2
scipy==1.7.1
torch==1.9.0+cu111
torchaudio==0.9.0
torchvision==0.10.0+cu111
tqdm==4.62.2
typing-extensions==3.10.0.2
解决方案
最简单的解决方案是设置int_classes = int
而不是从_six
.
推荐阅读
- asp.net-core - 如何在 Unix 上的 Service Fabric 应用程序中从 C# 获取机密证书?
- r - R:根据时间范围删除满足条件的行
- typescript - 我怎样才能实现 O = Omit
使 O & K 与 T 兼容 - javascript - 有没有办法在 React Native 中获取 UST 的实际时间(忽略用户设置)?
- javascript - 将 onScroll 钩子添加到第三方 reactJS 组件
- mongodb - Mongodb 检索记录 - 后缺少名称。操作员
- html - 如何更改 f:ajax 的 HTML 元素
- sql - 具有聚合的 Oracle Pivot
- java - weblogic12.2.1.3的apache poi ppt问题
- c# - 如何按角色获取页面权限