服务器导入torch.nn失败：ModuleNotFoundError问题求助

问题重现 #

在大学服务器环境中导入torch.nn时触发以下错误：

ModuleNotFoundError: No module named 'torch._prims_common'

已尝试新建虚拟环境、卸载重装torch/torchvision均无效，且无法正常查看torch版本。

解决方案 #

1. 彻底清理torch安装残留 #

卸载所有torch相关包并删除残留目录：

# 强制卸载所有torch相关组件
pip uninstall -y torch torchvision torchaudio
# 删除用户目录下的torch残留文件夹（Windows）
rmdir /s /q %USERPROFILE%\AppData\Roaming\Python\Python311\site-packages\torch
# Linux/macOS
rm -rf ~/.AppData/Roaming/Python/Python311/site-packages/torch

2. 确认Python与torch版本兼容性 #

先查看当前Python版本：

python --version

• Python 3.11需搭配torch 2.0.0及以上版本
• 若服务器有GPU，需同步确认CUDA版本（nvcc --version），选择对应兼容的torch版本

3. 重新完整安装torch #

根据环境选择对应安装命令：

• CPU版本：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

• GPU版本（以CUDA 11.8为例）：

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

安装完成后验证：

python -c "import torch; print(torch.__version__)"

若能正常输出版本，再尝试import torch.nn即可。

4. 规避多Python环境冲突 #

• 确保激活的是自己创建的虚拟环境，用以下命令确认当前Python路径：

  # Linux/macOS
  which python
  # Windows
  where python
  

• 若服务器支持conda，推荐用conda创建隔离环境安装，稳定性更高：

  conda create -n torch_env python=3.11
  conda activate torch_env
  # CUDA 11.8版本安装
  conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
  

额外注意 #

若服务器存在网络限制，导致安装包下载不完整，可切换至校内镜像源（若有）或使用代理完成安装。

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