在 Python 中使用 PyTorch 时,经常会碰到 no module named torch 的错误。本文总结了此错误出现的主要原因及对应的解决方法。

原因一:没有安装 PyTorch 

想在 Python 中调用 PyTorch,首先得安装它。可以在终端运行:

pip install torch torchvision torchaudio

来安装 PyTorch 和相关库。

原因二:PyTorch 版本不对  

如果安装了 PyTorch 的旧版本,但代码中需要较新版本的 API,也会出现此错误。解决方法是升级 PyTorch 到最新版。

原因三:环境变量没设置好 

有时虽然 PyTorch 安装成功,但环境变量没有正确设置,Python 不知道该如何找到 PyTorch。可以重新激活虚拟环境,或手动添加 PyTorch 的安装路径到 ENV PYTHONPATH。

原因四:使用的不是 PyTorch 所在环境 

如果系统有多个 Python 环境,且 PyTorch 只在某个环境中安装,而你运行代码使用的不是那个环境,也会报此错。解决方法有:
- 激活 PyTorch 所在环境后再运行代码
- 在当前环境也安装 PyTorch 
- 将 PyTorch 安装为全局环境

原因五:忘记 import torch 

这个原因比较尴尬,直接在 Python 代码开头忘记添加 import torch 导入语句,会报 no module named 'torch' 错误。

来源地址：https://blog.csdn.net/devid008/article/details/130280246