解决No module named ’torch._six‘问题

前言

最近升级的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_classes):
	# ...

if isinstance(root, str): 
	...

---

方法三

提高 torchvision 版本

conda install torchvision=0.15.1 -c pytorch -c nvidia

---

后续遇到其他类似问题继续更新