AI4Finance-Foundation / AI4Finance-Foundation/FinRL
How to use GPU
Ouverte
help_wanted
- Langage dominant
- Jupyter Notebook
- Étoiles
- 16.3k
- Forks
- 3.5k
- Métriques de merge des PR
- Aucune PR mergée en 30 j
Description
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 ?
Guide de contribution
Aucun guide de contribution indexé pour ce dépôt
Évaluation
Cette issue n'a pas encore été évaluée.