[env]pytorch_lightning和pytorch对应版本问题

问题

安装pytorch_lightning库，import pytorch_lightning as pl，报错OSError: libtorch_hip.so: cannot open shared object file: No such file or directory

错误截图如下：

 解决方法

原因

直接使用pip安装pytorch_lightning会安装最新版本的库，且自动更新环境中的torch版本，导致torch和cuda版本不兼容，无法使用torch。

重装torch后，因为torch版本与pytorch_lightning版本不兼容，无法使用pytorch_lightning出现上述错误。

方法

• 确定当前环境中的torch版本及python版本

import torch
import sys

torch_version = torch.__version__
python_version = sys.version

print("Torch version:", torch_version)
print("Python version:", python_version)

• 查找与之兼容的pytorch_lightning版本

 更多版本兼容情况，可以在官方文档查看 

pytorch_lightning官方文档https://lightning.ai/docs/pytorch/latest/versioning.html#pytorch-support

• 安装指定版本

pip install pytorch_lightning==1.8.0