No module named 'tensorflow.contrib'

[brunodbrazil]

I am having this problem when running on Windows. can anybody help me?

(base) C:\Users\XXXXXX>conda activate WAV

(WAV) C:\Users\XXXXXX>cd C:\Users\XXXXXX\wav

(WAV) C:\Users\XXXXXX\wav>python inference.py --checkpoint_path checkpoint/wav2lip.pth --face video/drive.mp4 --audio audio/audio.wav
2020-09-10 05:14:40.379812: W tensorflow/stream_executor/platform/default/dso_loader.cc:59] Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found
2020-09-10 05:14:40.383346: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
Traceback (most recent call last):
  File "inference.py", line 3, in <module>
    import scipy, cv2, os, sys, argparse, audio
  File "C:\Users\XXXXXX\wav\audio.py", line 7, in <module>
    from hparams import hparams as hp
  File "C:\Users\XXXXXX\wav\hparams.py", line 1, in <module>
    from tensorflow.contrib.training import HParams
ModuleNotFoundError: No module named 'tensorflow.contrib'

(WAV) C:\Users\XXXXXX\wav>

[KevinZzz35]

do not use tensorflow 2+

[prajwalkr]

Please install as per the requirements.txt and re-open the issue if you still have problems.

[brunodbrazil]

Please install as per the requirements.txt and re-open the issue if you still have problems.

I do this, but don't work.

[prajwalkr]

Please paste here the output of this command:

pip freeze | grep tensorflow

[rebotnix]

Please install as per the requirements.txt and re-open the issue if you still have problems.

I do this, but don't work.

It seems that your cudart is not loading:

"Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found...."

You are on windows, right? Make sure that cuda is correctly installed and search path are correct. I think that is the issue.

[prajwalkr]

Re-open if needed.