安装pytorch-lightning的坑
-
不能直接使用
pip install pytorch-lightning,这样会直接替换掉原来torch的版本。
torch安装方式一定要与pytorch-lightning安装方式一致。
应该:pip install pytorch-lightning==版本名 -
使用
pip install pytorch-lightning == 1.0.0安装版本过低会导致错误:TypeError: init() got an unexpected keyword argument ‘任意值’
解决方案:pip install pytorch-lightning==1.2.0