Linux Mint环境下导入PyTorch卡在torch.torch_version的问题求助

问题概况 #

在Linux Mint系统的Python虚拟环境（venv）中运行PyTorch代码时，import torch陷入无限等待，此前导入耗时最多10秒。测试代码如下：

import torch
x = torch.rand(5, 3)
print(x)

通过python -v -c "import torch"排查，进程卡在：

import 'torch.torch_version' # <_frozen_importlib_external.SourceFileLoader object at 0x7ec23812f470>

已尝试重启Jupyter内核、重载VSCode、外部文件运行、切换内核、降级torch至2.3.1版本，均未解决。环境信息：Python 3.12.3，虚拟环境包含大量依赖包。

可行解决方案 #

1. 排查依赖冲突 #

虚拟环境中大量依赖可能与PyTorch存在版本冲突，尤其是setuptools、wheel或PyTorch底层依赖库：

• 导出当前依赖列表：pip freeze > requirements.txt
• 创建全新虚拟环境，仅安装PyTorch核心包测试：

  deactivate
  rm -rf ./your_venv
  python3.12 -m venv new_venv
  source new_venv/bin/activate
  # 根据硬件选择CPU/CUDA版本，示例为CPU版
  pip install torch==2.3.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
  

  若导入正常，再逐步添加其他依赖，定位冲突包。

2. 清理PyTorch残留文件 #

旧版本残留文件可能导致加载异常：

• 卸载当前PyTorch：pip uninstall -y torch torchvision torchaudio
• 删除虚拟环境中PyTorch相关目录：

  rm -rf $VIRTUAL_ENV/lib/python3.12/site-packages/torch*
  rm -rf $VIRTUAL_ENV/lib/python3.12/site-packages/torchvision*
  rm -rf $VIRTUAL_ENV/lib/python3.12/site-packages/torchaudio*
  

• 从官方源重新安装对应版本的PyTorch。

3. 验证Python版本兼容性 #

Python 3.12属于较新版本，部分PyTorch版本适配性可能不足：

• 降级Python至3.11（PyTorch对该版本支持更成熟），重新创建虚拟环境并安装PyTorch测试。

4. 补充系统底层依赖 #

Linux Mint缺少必要系统库可能导致PyTorch加载卡住：

• 安装PyTorch所需系统依赖：

  sudo apt update
  sudo apt install libopenblas-dev libomp-dev libopencv-dev
  

• 若使用CUDA版本，确保CUDA驱动与PyTorch版本匹配，重新验证CUDA环境变量配置。

5. 检查文件系统权限 #

虚拟环境或PyTorch安装目录权限异常可能导致加载失败：

• 查看虚拟环境目录权限：ls -ld ./your_venv
• 若权限不足，修改归属：sudo chown -R $USER:$USER ./your_venv

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