解决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