pytorch_lightning:ImportError: initialization failed报错
本人报错的包是pytorch_lightning,但是经过各种上网搜索后得到的解决方法其实是很扯淡的,也是让人痛苦的,并且这个方法可以通用于任何包的ImportError: initialization failed报错(如果你真的是这个错误的话)
注意:这个解决方法是你在尝试了网上各种方法后都没法解决后可能去试一试的方法
检查你的torch版本和CUDA版本!!
简单质朴而通透
我跑的代码推荐torch版本是1.9.0 ,然而这个上古机器的CUDA却是10.2
所以如果你走投无路了,就去查看一下的你的CUDA版本是不是太老以至于配不上你的torch版本
import torch
#import pytorch_lightning as pl
if torch.cuda.is_available():
print("CUDA is available")
else:
print("CUDA is not available")
print(torch.version.cuda)
print(torch.__version__)
输出:
CUDA is available
10.2
1.9.0+cu102
如果对不上,就去更新的CUDA吧(如何更新CUDA教程一大堆)
比如这个更新CUDA的教程
友情附上pytorch的官网,去这里用他的命令行下载torch,要注意区分GPU和CPU