I am trying to import Nvidia NeMo core package into my Google Colaboratory notebook (connected to local runtime) when this error pops up: ModuleNotFoundError: No module named 'pytorch_lightning.callbacks.base'
These were the codes:
BRANCH = 'r1.21.0'
!python -m pip install git+https://github.com/NVIDIA/NeMo.git@$BRANCH#egg=nemo_toolkit[all]
# and then
import nemo
import nemo.collections.asr as nemo_asr
These were the logs:
-----------------------------------------------------------------------
ModuleNotFoundError Traceback (most recent call last)
Cell In[3], line 5
2 import nemo
3 # NeMo's ASR collection - this collections contains complete ASR models and
4 # building blocks (modules) for ASR
----> 5 import nemo.collections.asr as nemo_asr
File ~\NeMo\nemo\collections\asr\__init__.py:15
1 # Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
(...)
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
---> 15 from nemo.collections.asr import data, losses, models, modules
16 from nemo.package_info import __version__
18 # Set collection version equal to NeMo version.
File ~\NeMo\nemo\collections\asr\models\__init__.py:16
1 # Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
(...)
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
15 from nemo.collections.asr.models.asr_model import ASRModel
---> 16 from nemo.collections.asr.models.classification_models import EncDecClassificationModel
17 from nemo.collections.asr.models.clustering_diarizer import ClusteringDiarizer
18 from nemo.collections.asr.models.ctc_bpe_models import EncDecCTCModelBPE
File ~\NeMo\nemo\collections\asr\models\classification_models.py:28
25 from pytorch_lightning import Trainer
26 from torchmetrics.regression import MeanAbsoluteError, MeanSquaredError
---> 28 from nemo.collections.asr.data import audio_to_label_dataset
29 from nemo.collections.asr.models.asr_model import ASRModel, ExportableEncDecModel
30 from nemo.collections.asr.parts.preprocessing.features import WaveformFeaturizer
File ~\NeMo\nemo\collections\asr\data\audio_to_label_dataset.py:15
1 # Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
(...)
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
---> 15 from nemo.collections.asr.data import audio_to_label
16 from nemo.collections.asr.data.audio_to_text_dataset import convert_to_config_list, get_chain_dataset
19 def get_classification_label_dataset(featurizer, config: dict) -> audio_to_label.AudioToClassificationLabelDataset:
File ~\NeMo\nemo\collections\asr\data\audio_to_label.py:22
19 import torch
20 import webdataset as wd
---> 22 from nemo.collections.asr.parts.preprocessing.segment import available_formats as valid_sf_formats
23 from nemo.collections.common.parts.preprocessing import collections
24 from nemo.core.classes import Dataset, IterableDataset
File ~\NeMo\nemo\collections\asr\parts\preprocessing\__init__.py:16
1 # Copyright (c) 2021, NVIDIA CORPORATION. All rights reserved.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
(...)
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
15 from nemo.collections.asr.parts.preprocessing.feature_loader import ExternalFeatureLoader
---> 16 from nemo.collections.asr.parts.preprocessing.features import (
17 STFT,
18 FeaturizerFactory,
19 FilterbankFeatures,
20 STFTExactPad,
21 STFTPatch,
22 WaveformFeaturizer,
23 )
24 from nemo.collections.asr.parts.preprocessing.perturb import (
25 AudioAugmentor,
26 AugmentationDataset,
(...)
39 register_perturbation,
40 )
41 from nemo.collections.asr.parts.preprocessing.segment import AudioSegment
File ~\NeMo\nemo\collections\asr\parts\preprocessing\features.py:47
44 from torch.autograd import Variable
45 from torch_stft import STFT
---> 47 from nemo.collections.asr.parts.preprocessing.perturb import AudioAugmentor
48 from nemo.collections.asr.parts.preprocessing.segment import AudioSegment
49 from nemo.utils import logging
File ~\NeMo\nemo\collections\asr\parts\preprocessing\perturb.py:50
47 from torch.utils.data import IterableDataset
49 from nemo.collections.asr.parts.preprocessing.segment import AudioSegment
---> 50 from nemo.collections.common.parts.preprocessing import collections, parsers
51 from nemo.utils import logging
53 # TODO @blisc: Perhaps refactor instead of import guarding
File ~\NeMo\nemo\collections\common\__init__.py:15
1 # Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
(...)
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
---> 15 import nemo.collections.common.callbacks
16 from nemo.collections.common import data, losses, parts, tokenizers
17 from nemo.package_info import __version__
File ~\NeMo\nemo\collections\common\callbacks\__init__.py:15
1 # Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
(...)
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
---> 15 from nemo.collections.common.callbacks.callbacks import LogEpochTimeCallback
File ~\NeMo\nemo\collections\common\callbacks\callbacks.py:17
14 import time
16 import numpy as np
---> 17 from pytorch_lightning.callbacks.base import Callback
18 from pytorch_lightning.utilities import rank_zero_only
20 # from sacrebleu import corpus_bleu
ModuleNotFoundError: No module named 'pytorch_lightning.callbacks.base'
Previously, I have managed to import the packages into my local machine but just recently, I had to reinstall my Anaconda. So I thought this might be due to some dependencies not being installed. However I have already installed pytorch lightning via 'pip install pytorch_lightning' and it was a success.
Any help on this, am I missing another import?