最近升级的pytorch的版本,升级到了2.0
但是加载之前的项目时出现了:No module named ’torch._six‘ 的问题
记录下解决办法
方法一
降低pytorch的版本,pytorch2.0 没有_six.py文件,导致无法找到
本人也尝试了将pytorch1.10版本下的_six.py文件拷贝到pytorch2.0下,无果
方法二
找到代码中调用torch._six的部分替换
情况一:
# from torch._six import *
int_classes = int
string_classes = str
情况二:
# if isinstance(root, torch._six.string_