torch.amp模块缺失报错求助及成功修复方案分享

问题现象 #

原本可在Jupyter Notebook中成功执行import torch，后续突然出现如下报错：

import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Users\graw\AppData\Roaming\Python\Python310\site-packages\torch\__init__.py", line 795, in <module>
    from torch.amp import autocast
ModuleNotFoundError: No module named 'torch.amp'

尝试过的无效操作 #

使用以下命令重新安装torch（原版本为torch-2.0.1+cu118），问题仍未解决：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

有效解决方法 #

将本地Python安装目录下的torch文件夹（路径示例：...\python 3.10.6\Lib\site-packages\torch），复制并替换到用户AppData目录下的torch路径（示例：C:\Users...\AppData\Roaming\Python\Python310\site-packages\torch），执行后问题解决。

内容的提问来源于stack exchange，提问作者Graw