导入 torch.fx ModuleNotFoundError：没有名为“torch.fx”的模块

Question

根据我的系统 GPU 兼容性为我的系统启用 torch 和 Cuda 后，每当我尝试运行任何需要在 GPU 上运行以启用系统的程序时，就会出现此错误。我无法找到任何解决方案。虽然我读到了有关创建另一个环境的内容，并且此错误将得到解决，但对我来说不起作用。请查找我的系统的详细信息。作为参考，我使用的是 Python 3.7.2。

图像包括 torch 和 Cuda 版本以及设备名称

NVIDIA Cuda编译器驱动详细信息

nvidia-smi 输出

任何人都可以帮助解决这个问题吗？谢谢

Answer 1

torch.fxPyTorch 1.8.0 中添加。检查发布帖子。您可能使用的是旧版本。从网站升级 pytorch 。

Answer 2

而不是使用

import torch.fx

您必须从 torch.fx 导入，如下所示：

from torch.fx import symbolic_trace

更多内容可以查看官方文档。