使用TensorFlow、Pytorch等深度学习框架时如何设置对OpenCV的使用

如题：

在使用深度学习框架时如果同时也在使用opencv那么有一些设置是需要设定的，第一个就是在python代码中设定禁止使用opencl：

cv2.ocl.setUseOpenCL(False)

 

 

 

 

 

 

opencl和cuda的基本功能一致：

 

opencl和cuda当年是作为同等定位的软件产品的，但是后来由于amd公司在这个领域上的落败已经很少见到opencl的产品了，而基本所有的深度学习框架都是基于cuda的，但是opencv却默认支持opencl的，如果一个应用中同时使用cuda和opencl就有可能造成一点的错误。在使用cuda的前提下为避免产生未知错误最好禁用opencl。

 

 

 

 

 

在pytorh中如果在数据集读取加载过程中使用opencv可能由于opencv的自动多线程可能会与pytorch的多进程dataloader造成互锁因此我们也往往需要将opencv的多线程设置关闭：

同样是在python文件中添加设置语句：

cv2.setNumThreads(0)

 

 

 

 

 

总的来说，在使用深度学习框架的时候如果同时使用opencv最好进行下面的设置：

在python文件最初的地方添加设置语句：

cv2.setNumThreads(0)
cv2.ocl.setUseOpenCL(False)

 

 

 

 

=============================================

 

 

 

 

参考：

https://www.cnblogs.com/yqs-0705/archive/2018/12/15/10122988.html

https://www.e-learn.cn/topic/3316419

https://zhuanlan.zhihu.com/p/377013846