当前位置:   article > 正文

Python 使用各种包 报错ImportError: initialization failed的可能解决方法之一 如import pytorch_lightning等

importerror: initialization failed

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__)

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11

输出:

CUDA is available
10.2
1.9.0+cu102
  • 1
  • 2
  • 3

如果对不上,就去更新的CUDA吧(如何更新CUDA教程一大堆)
比如这个更新CUDA的教程
友情附上pytorch的官网,去这里用他的命令行下载torch,要注意区分GPU和CPU

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/468673
推荐阅读
相关标签
  

闽ICP备14008679号