ModuleNotFoundError: No module named 'torch.fx'

PyTorch version is 1.7.0 and i try to tun this command:

python3 detect.py --source inference/images/horses.jpg --cfg cfg/yolor_p6.cfg --weights yolor_p6.pt --conf 0.25 --img-size 1280 --device 0

it gaved me this error:

from torch.fx import symbolic_trace

ModuleNotFoundError: No module named 'torch.fx'

In first, i install torch 1.8.0 but when i have “RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED - PyTorch”; so, i install torch 1.7.0 to solve this problem and it solved.

I read something like that: “torch.fx comes with torch 1.8.0” but i can not upgrade because of “RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED - PyTorch” error.

How can i solve this problem: ModuleNotFoundError: No module named 'torch.fx' ?