AI4Finance-Foundation / AI4Finance-Foundation/FinRL_Crypto

How to improve this model's train effeciency on GPU

Offen
#4 0 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
help wanted
Vorherrschende Sprache
Python
Sterne
201
Forks
88
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

I tried to run the part of cpcv of this model on Google Colab Pro+ with A100 GPU but it only uses 1.7GB over 40GB GPU RAM. No matter how I changed the batch size, worker_num of the GPU, thread_num of the CPU. It seems to me that there is a bottleneck that restrains the GPU performance and train speed. Can anyone help with this problem?

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.