记录一下自己遇到的一个坑
背景:在服务器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包的冲突。
参考链接: