remove cuda debugging

ml6team · Oct 5, 2023 · 0a255da · 0a255da
1 parent caa649a
commit 0a255da
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 4 deletions.
diff --git a/components/caption_images/src/main.py b/components/caption_images/src/main.py
@@ -13,7 +13,6 @@
 
 logger = logging.getLogger(__name__)
 
-os.environ['CUDA_LAUNCH_BLOCKING'] = "1"
 os.environ['TORCH_CUDNN_V8_API_DISABLED'] = "1"
 
 

diff --git a/components/embed_images/src/main.py b/components/embed_images/src/main.py
@@ -13,7 +13,6 @@
 
 logger = logging.getLogger(__name__)
 
-os.environ['CUDA_LAUNCH_BLOCKING'] = "1"
 os.environ['TORCH_CUDNN_V8_API_DISABLED'] = "1"
 
 

diff --git a/components/segment_images/src/main.py b/components/segment_images/src/main.py
@@ -14,7 +14,6 @@
 
 logger = logging.getLogger(__name__)
 
-os.environ['CUDA_LAUNCH_BLOCKING'] = "1"
 os.environ['TORCH_CUDNN_V8_API_DISABLED'] = "1"
 
 

diff --git a/examples/pipelines/datacomp/components/detect_text/src/main.py b/examples/pipelines/datacomp/components/detect_text/src/main.py
@@ -23,7 +23,6 @@
 
 logger = logging.getLogger(__name__)
 
-os.environ["CUDA_LAUNCH_BLOCKING"] = "1"
 os.environ["TORCH_CUDNN_V8_API_DISABLED"] = "1"