[env]pytorch_lightning和pytorch对应版本问题
直接使用pip安装pytorch_lightning会安装最新版本的库,且自动更新环境中的torch版本,导致torch和cuda版本不兼容,无法使用torch。重装torch后,因为torch版本与pytorch_lightning版本不兼容,无法使用pytorch_lightning出现上述错误。pytorch_lightning官方文档。更多版本兼容情况,可以在官方文档查看。
原创
2023-08-11 14:33:31 ·
18281 阅读 ·
4 评论