albumentations包无法导入

记录一下自己遇到的一个坑

背景:在服务器linux系统下,使用pytorch搭建深度学习网络,准备使用Albumentations包做数据扩增,但是在使用pip安装albumentations包之后,导入该包无限等待,无法继续往下进行。

在网上搜索问题之后,尝试了各种方法:

1)尝试了不同的albumentations包

2)尝试不同版本的python

3)尝试不同版本的pytorch

都会出现相同的问题,再重新创环境后发现,只安装albumentations包导入是没有问题的。但是,在安装pytorch之后,包的导入就出了问题。

之后在网上看到一个答案是说pytorch与opencv-python包的版本问题会发生冲突,只要先导入pytorch包或者降低opencv-python的版本就可以,我就想到albumentations包的初始化中导入了cv2,于是我尝试先导入pytorch,再导入albumentations包,问题解决!

降低opencv-python包版本的操作我没有尝试,估计也是可以顺利解决问题的,问题根源应该就是pytorch包与opencv-python包的冲突。

参考链接:

openCV与tensorflow、pytorch的版本冲突问题 - 番茄-- - 博客园

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值