Enabled OpenCL in OpenCV (#24)

* Enabled OpenCL in OpenCV * Added comment about OpenCL block to README * Added comment about OpenCL block to README
albumentations-team · Jul 10, 2018 · fa0fd0a · fa0fd0a
1 parent 9361d3a
commit fa0fd0a
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 8 deletions.
diff --git a/README.md b/README.md
@@ -129,5 +129,15 @@ Alternatively, you can start a web server that rebuilds the documentation
 automatically when a change is detected by running `make livehtml`
 
 
+## Comments
+In some systems, in the multiple GPU regime PyTorch may deadlock the DataLoader if OpenCV was compiled with OpenCL optimizations. Adding the following two lines before the library import may help. For more details [https://github.com/pytorch/pytorch/issues/1355](https://github.com/pytorch/pytorch/issues/1355) 
+
+
+
+```python
+cv2.setNumThreads(0)	
+cv2.ocl.setUseOpenCL(False)
+```
+
 ### Thanks:
 Special thanks to [@creafz](https://github.com/creafz) for refactoring, documentation, tests, CI and benchmarks. Awesome work!
diff --git a/albumentations/augmentations/functional.py b/albumentations/augmentations/functional.py
@@ -3,9 +3,6 @@
 
 import cv2
 
-cv2.setNumThreads(0)
-cv2.ocl.setUseOpenCL(False)
-
 import numpy as np
 from scipy.ndimage.filters import gaussian_filter
 

diff --git a/albumentations/augmentations/transforms.py b/albumentations/augmentations/transforms.py
@@ -2,9 +2,6 @@
 
 import cv2
 
-cv2.setNumThreads(0)
-cv2.ocl.setUseOpenCL(False)
-
 import numpy as np
 
 from ..core.transforms_interface import to_tuple, DualTransform, ImageOnlyTransform

diff --git a/albumentations/core/transforms_interface.py b/albumentations/core/transforms_interface.py
@@ -1,7 +1,5 @@
 import numpy as np
 import cv2
-cv2.setNumThreads(0)
-cv2.ocl.setUseOpenCL(False)
 
 __all__ = ['to_tuple', 'BasicTransform', 'DualTransform', 'ImageOnlyTransform']