Pytorch-lightning: ImportError: cannot import name 'WandbLogger' from 'pytorch_lightning.loggers'

[braindotai]

🐛 Bug

I tried importing wandb logger from pytorch_lightning.loggers and got import error.

Code sample

from pytorch_lightning.loggers import WandbLogger

Environment

• PyTorch Version (e.g., 1.0): 1.4.0
• OS (e.g., Linux): Windows 10
• How you installed PyTorch (conda, pip, source): pip
• Python version: 3.7.7
• CUDA/cuDNN version: Cuda compilation tools, release 10.1, V10.1.105
• GPU models and configuration: 1050ti

[github-actions[bot]]

Hi! thanks for your contribution!, great first issue!

[olineumann]

What is the exact error message? Which lightning version is installed?

[awaelchli]

from pytorch_lightning.loggers.wandb import WandbLogger also doesn't work?

[lezwon]

You need to install wandb package first.
pip install wandb

[Borda]

You may also install all extra libs pip install -r requirements-extra.txt
@braindotai feel free to re-open if needed :rabbit:

[lezwon]

Guess it would be helpful if there was a error displayed suggesting to install the logger package while importing. I had this same issue while using comet logger and wasn't sure what was wrong.

[braindotai]

I had this same issue while using comet logger and wasn't sure what was wrong.

Actually I opened my code in vs code, and after that, I installed wandb and did the logging. It needed a restart to the editor to setup wandb properly with my editor command line, and after the restart, the error was gone indeed. I guess that's what happened with you as well.

The issue is solved. Thanks to everyone for the quick response.