Как исправить проблему с GPU при обучении модели? - PullRequest
0 голосов
/ 30 апреля 2019

Я строю модель face recognition, для face alignment я использую MTCNN с набором данных с 90,56,342 изображениями и 9540 отдельными объектами.

Ниже приведена конфигурация моей машины:

GPU: MSI GTX 1060 6 ГБ

Процессор: 16 ГБ, DD4

Тактовая частота: 2,8 ГГц

Процессор: Процессор Intel Xeon.

После запуска Face Alignment, обработка была остановлена, и через GPU невозможно определить.

Я пытался уменьшить GPU utilization fraction, но безрезультатно. Я попытался построить другую маленькую модель, но снова выдает ту же ошибку.

Я установил все стабилизированные версии

CUDA: 9,2

cudNN: 7,3

NVIDIA-DRIVER_version: 430,13

питон: 3.6.8

Тензор потока-GPU: 1,13

Я проверил Power Supply Unit получения необходимого источника питания при обработке модели.

Блок питания: AC 230V 3A

Я установил для temperature графического процессора менее 40 C.

...