解决ModuleNotFoundError: No module named ‘torch.fx‘

运行yolo v5 train
python train.py

报错
ModuleNotFoundError: No module named ‘torch.fx’

torch版本不匹配
目前版本torchu1.7

#卸载PyTorch
pip uninstall torch

再安装

Python -m pip install torch -i https://mirrors.aliyun.com/pypi/simple/

python -m pip是什么?

首先，python -m pip会使用您指定为python的Python解释器来执行pip。因此，/usr/bin/python3.7 -m pip表示您正在执行位于/usr/bin/python3.7的解释器的pip。如果您不熟悉这个标志以及它是如何工作的，您可以阅读有关-m的文档(它非常方便)。

(result, consumed) = self._buffer_decode(data, self.errors, final)

UnicodeDecodeError: ‘utf-8’ codec can’t decode byte 0xd3 in position 357: invalid continuation byte

查看CUDA版本
(base) menghan@menghan-NH5x-7xRCx-RDx:~/xiazai/PyCharm-commUnity-2022.1/bin$ cat /usr/local/cuda/version.txt
CUDA Version 9.0.176

来源地址：https://blog.csdn.net/weixin_48888528/article/details/127834968