Example not running - ModuleNotFound

[EceChaik]

Hi everyone,

I just went through the installation instructions in the wiki, and after installing everything, I run the proposed example python3 run_drone_control.py --train 0

But I get the following error: python3 run_drone_control.py --train 0 2021-03-19 14:14:08.561713: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0 Traceback (most recent call last): File "run_drone_control.py", line 12, in from stable_baselines import logger File "/home/dc4204/anaconda3/envs/Flightmare/lib/python3.6/site-packages/stable_baselines/init.py", line 6, in from stable_baselines.deepq import DQN File "/home/dc4204/anaconda3/envs/Flightmare/lib/python3.6/site-packages/stable_baselines/deepq/init.py", line 1, in from stable_baselines.deepq.policies import MlpPolicy, CnnPolicy, LnMlpPolicy, LnCnnPolicy File "/home/dc4204/anaconda3/envs/Flightmare/lib/python3.6/site-packages/stable_baselines/deepq/policies.py", line 2, in import tensorflow.contrib.layers as tf_layers ModuleNotFoundError: No module named 'tensorflow.contrib'

Any help appreciated, thanks