FinRL icon indicating copy to clipboard operation
FinRL copied to clipboard

Published 20 hours ago • verified publisher icon AI4Finance-Foundation

How to use GPU

Open avatarproject123 opened this issue 1 year ago • 3 comments

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 Logging to results/a2c

how to use gpu ?

avatarproject123 avatar Jun 01 '23 10:06 avatarproject123