Can we use it with Tenserflow 2.3.1 ?

[deepakraina99]

When I run the enjoy.py file, I get the following error:

ModuleNotFoundError: No module named 'tensorflow.contrib'

After googling this error, I found that this only works with Tensorflow 1.x.x versions

[araffin]

Hello,

As mentioned in the doc, SB2 officially only support TF1. There is an unofficial port, see hill-a/stable-baselines#1012 .
Our next major version, Stable-Baselines3 supports PyTorch.

[deepakraina99]

Thanks for the information.

[amexn-me]

Have you tried any workaround?

[deepakraina99]

No

[amexn-me]

I see. I tried downgrading TF to 1.15 like araffin said and it works for now.