如何解决Python中PyTorch的ModuleNotFoundError问题

执行以下Python代码时出现ModuleNotFoundError：

import torch
device = "cuda" if torch.cuda.is_available() else "cpu"
print(device)

错误信息：

ModuleNotFoundError                       Traceback (most recent call last)
Cell In[1], line 1
----> 1 import torch
      2 device = "cuda" if torch.cuda.is_available() else "cpu"
      3 print(device)

ModuleNotFoundError: No module named 'torch'

已通过以下命令安装PyTorch：

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

且在CUDA的lib文件夹中找到torch，但程序无法从该路径加载模块。

• 检查Python与pip的路径匹配
  运行以下命令确认python3和pip3的路径是否一致：

  which python3
  which pip3
  

  若路径不同，说明pip安装的包不在当前Python环境的搜索路径中，改用以下命令重新安装：

  python3 -m pip install torch --index-url https://download.pytorch.org/whl/cu118
  

• 验证torch的安装路径与Python搜索路径

  1. 查看当前Python的模块搜索路径：

     import sys
     print(sys.path)
     

  2. 查看torch的实际安装位置：

     pip3 show torch
     

  若torch的安装路径不在sys.path中，可在代码中手动添加路径后再导入：

  import sys
  sys.path.append("/path/to/torch/folder") # 替换为实际的torch所在路径
  import torch
  

• 检查虚拟环境状态
  如果使用了虚拟环境，确认安装torch时是否激活了该环境。若未激活，包会被安装到全局环境，虚拟环境无法访问。激活目标虚拟环境后重新执行安装命令即可。

• 彻底卸载重装torch

  1. 卸载现有torch：

     pip3 uninstall -y torch
     

  2. 清理pip缓存：

     pip3 cache purge
     

  3. 重新安装：

     python3 -m pip install torch --index-url https://download.pytorch.org/whl/cu118
     

• 确认CUDA版本兼容性
  运行nvcc --version查看本地CUDA版本，确保为11.8（与安装命令中的cu118对应）。若版本不匹配，需升级/降级CUDA，或更换对应CUDA版本的torch安装命令。

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