我一直在建立face recognition
模型,对于face alignment
,我正在将MTCNN
与具有90,56,342
图像和9540
个个体的数据集一起使用。
以下是我的机器配置:
GPU:MSI GTX 1060 6gb
CPU:16GB DD4
时钟速度:2.8 GHz
处理器:Intel Xeon处理器。
Face Alignment
启动后,处理停止,无法确定GPU。
我尝试减少GPU utilization fraction
,但没有用。
我尝试构建另一个小模型,但同样会给出相同的错误。
我已经安装了所有稳定版的
CUDA:9.2
cudNN:7.3
NVIDIA-DRIVER_version:430.13
python:3.6.8
tensorflow-gpu:1.13
我检查了Power Supply Unit
,以便在处理模型时获得所需的电源。
PSU:AC 230V 3A
我已将GPU的temperature
控制为小于40 C
。