Почему optuna нагружает мой процессор, а не графический процессор, когда для устройства установлено значение «cuda»? ⇐ Python

[Anonymous]

Я использую optuna для настройки гиперпараметров, и хотя на моем устройстве установлено значение «cuda», и оно фактически работает на cuda, так как процессору требуется 40 минут для завершения 10 эпох, в то время как в настоящее время это занимает всего 6 минут на прохождение 30 эпох. Это означает, что моя программа использует графический процессор. Однако я обнаружил, что процессор загружен на 100 %, а мой графический процессор практически не используется программой.

Вот мои характеристики аппаратного и программного обеспечения:

Код: Выделить всё

Hardware specifications:
Lenovo Legion 5 2022
Ryzen 7 6800H
NVIDIA RTX 3060 TDP 140W
16 GB DDR5 RAM 4800 Mhz
1TB PCIE Gen 4 SSD
Optimus disabled (NVIDIA Dgpu only)

Код: Выделить всё

Software specifications:
python 3.10.9
conda 23.3.1
optuna 3.6.0 conda-forge
optuna-dashboard 0.15.1 conda-forge

Мне не удалось придумать, как решить эту проблему.

Подробнее здесь: https://stackoverflow.com/questions/785 ... et-to-cuda