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

Author：小鹏AI
link：https://blog.csdn.net/qq_38973721/article/details/130947938

---

最近升级的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

最终解决办法

注意，此时的 torch 和 ltp 版本
ltp version is 4.1.5.post2

C:\Users\54867>pip show ltp
Name: ltp
Version: 4.1.5.post2
Summary: Language Technology Platform
Home-page: https://github.com/HIT-SCIR/ltp
Author: Yunlong Feng
Author-email: ylfeng@ir.hit.edu.cn
License: UNKNOWN
Location: d:\anaconda\lib\site-packages
Requires: torch, packaging, pygtrie, transformers
Required-by:

torch version is 2.2.1

C:\Users\54867>pip show torch
Name: torch
Version: 2.1.1
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: packages@pytorch.org
License: BSD-3
Location: d:\anaconda\lib\site-packages
Requires: jinja2, typing-extensions, filelock, networkx, fsspec, sympy
Required-by: ltp, ltp-core

将 from torch._six import string_classes 注销

import torch
# from torch._six import string_classes
import collections.abc as container_abcs

手把手教你解决ModuleNotFoundError: No module named ‘torch._six‘