训练模型时如何解决GPU问题?

时间:2019-04-30 06:54:49

标签: python deep-learning computer-vision gpu

我一直在建立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

0 个答案:

没有答案