ModuleNotFoundError：沒有名為“torch.utils.benchmark”的模塊

Question

我一直在嘗試使用火炬基准模塊來分析代碼。 驗證所有必需的導入都已到位：

import torch
import torchvision
import torch.utils.benchmark 

此外，這些是安裝的版本：

torch ->  1.4.0
torchvision -> 0.2.1
tensorflow ->  1.12.0
tensorflow-gpu -> 1.12.0
tensorboard -> 2.5.0

但我仍然收到錯誤： ModuleNotFoundError: No module named 'torch.utils.benchmark' while importing torch.utils.benchmark 。 根本原因可能是什么？

Answer 1

你試過升級手電筒嗎？ 我看到您使用的是 1.4.0，但目前可以使用 1.9.0。

pip install torch --upgrade

我已經按照您在 1.9.0 中描述的那樣測試了模塊的導入，並且沒有收到任何錯誤。