yolov5 云端训练笔记 ModuleNotFoundError: No module named 'torch.fx'

[一只会飞的羊]

pip install -r /root/yolov5-3.0/requirements.txt

运行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/

 

 安装成功，问题解决