disable nvfuser

Signed-off-by: ericharper <[email protected]>
NVIDIA · Mar 15, 2022 · 273f623 · 273f623
1 parent a18e493
commit 273f623
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/nlp/models/language_modeling/megatron_gpt_model.py b/nemo/collections/nlp/models/language_modeling/megatron_gpt_model.py
@@ -106,7 +106,7 @@ def __init__(self, cfg: DictConfig, trainer: Trainer):
         self._validate_trainer()
 
         # used in NVIDIA NGC PyTorch containers
-        self._enable_nvidia_optimizations()
+        # self._enable_nvidia_optimizations()
 
         if self.cfg.get('use_cpu_initialization', False) is False:
             torch.cuda.set_device(trainer.local_rank)