PyTorch Lightning导入报错求助：找不到libtorch_hip.so文件

错误原因 #

libtorch_hip.so是PyTorch针对AMD GPU（ROCm平台）的专属库文件，出现找不到该文件的错误，核心原因要么是PyTorch安装版本与当前硬件/环境不匹配，要么是依赖包版本冲突导致加载错误。

具体解决步骤 #

• 检查PyTorch适配性
  若你的机器是NVIDIA GPU或CPU，大概率误装了ROCm版本的PyTorch。执行以下操作修复：

  1. 卸载现有PyTorch：

     pip uninstall torch -y
     

  2. 重新安装对应版本：
     • NVIDIA GPU（以CUDA 11.8为例）：

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

     • CPU版本：

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

• 排查依赖版本冲突
  pytorch-forecasting 1.0.0对pytorch-lightning的版本可能存在兼容限制，尝试降级pytorch-lightning到兼容版本：

  pip install pytorch-lightning==1.9.5
  

  安装完成后重新测试import lightning.pytorch as pl。

• AMD GPU环境配置（若使用AMD显卡）
  确保ROCm环境已正确安装，且库路径被添加到环境变量：

  export LD_LIBRARY_PATH=/opt/rocm/lib:$LD_LIBRARY_PATH
  

  注意根据你的ROCm实际安装路径调整上述命令中的路径。

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