AI4Finance-Foundation / AI4Finance-Foundation/ElegantRL

请教一个多GPU训练的问题

Aperta
#278 1 commento 0 reazioni 0 assegnatari Vedi su GitHub
dicussion
Lingua principale
Python
Stelle
4.4k
Fork
978
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

我看文档说支持多GPU训练,比如tutorial_BipedalWalker-v3.py里面的例子
args.learner_gpus = [0, 1, 2, 3]
train_agent_multiprocessing(args)
然后我看了一下源码,貌似learner_gpus这个参数后来根本没有用过
train_agent_multiprocessing这里面好像也没有对gpu_id分配的管理
不知道我是不是漏了关键代码了,请问这个多GPU训练是怎么实现的呢?谢谢。

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.