Error when wandb is installed

System Summary

Pop!_OS 20.04 Pytorch: 1.5.1 Transformers: 3.0.2 Tokenizers: 0.8.1rc1 Python: 3.7.6 Pretrained Model: GPT2 Pretrained Tokenizer: GPT2

Question

Training without wandb works fine but when I pip install wandb and change nothing else in my code, whenever I go to run training I get the following error:

I0821 15:44:17.531560 46912496399424 file_utils.py:39] PyTorch version 1.5.1 available.
I0821 15:44:21.471980 46912496399424 file_utils.py:55] TensorFlow version 2.0.0 available.
Traceback (most recent call last):
  File "run_finetune_gpt2.py", line 5, in <module>
    from transformers import TrainingArguments, Trainer, GPT2Tokenizer
  File "/path/to/venv/my-venv/lib/python3.6/site-packages/transformers/__init__.py", line 158, in <module>
    from .trainer_utils import EvalPrediction, set_seed
  File "/path/to/venv/my-venv/lib/python3.6/site-packages/transformers/trainer_utils.py", line 11, in <module>
    import wandb
  File "/path/to/venv/my-venv/lib/python3.6/site-packages/wandb/__init__.py", line 60, in <module>
    from wandb.apis import InternalApi, PublicApi, CommError
  File "/path/to/venv/my-venv/lib/python3.6/site-packages/wandb/apis/__init__.py", line 116, in <module>
    from .public import Api as PublicApi
  File "/path/to/venv/my-venv/lib/python3.6/site-packages/wandb/apis/public.py", line 28, in <module>
    from wandb.summary import HTTPSummary
  File "/path/to/venv/my-venv/lib/python3.6/site-packages/wandb/summary.py", line 15, in <module>
    from wandb.meta import Meta
  File "/path/to/venv/my-venv/lib/python3.6/site-packages/wandb/meta.py", line 6, in <module>
    import pynvml
  File "/cm/local/apps/cuda/libs/current/pynvml/pynvml.py", line 1671
    print c_count.value
                ^
SyntaxError: Missing parentheses in call to 'print'. Did you mean print(c_count.value)?

Any thoughts?

[aclifton314]

@borisdayma that link solved my problem! Thanks for your help!

[borisdayma]

I ran it in colab: see notebook

There does not seem to be any issue related to wandb. Maybe try my previous link as it may be due to issues in your local environment.