I have cuda installed. I have given "device": "cuda" in params but I am still getting using cpu device.
A2C_PARAMS = {"n_steps": 5, "ent_coef": 0.01, "learning_rate": 0.0005, "device": "cuda"}
output:
{'n_steps': 5, 'ent_coef': 0.01, 'learning_rate': 0.0005, 'device': 'cuda'}
Using cpu device
Loggin...