YOLO - tensorflow可以在cpu上运行，但不能在gpu上运行

Question

我已经使用我的GPU - Nvidia 1060 3 3Gb训练过的模型使用了YOLO检测，一切工作正常。

现在，我正在尝试使用参数生成自己的模型--gpu 1.0。Tensorflow可以看到我的图形处理器，因为我可以在开始时读到这些通信："name: GeForce GTX1060大调:6小调:1 memoryClockRate(GHz)：1.6705“"totalMemory: 3.00GiB freeMemory: 2.43GiB”

无论如何，后来，当程序加载数据并试图开始学习时，我得到了以下错误：“无法从设备分配832.51M (872952320字节)：CUDA_ERROR_OUT_OF_MEMORY”

我检查过它是否尝试使用我的另一个gpu (英特尔630)，但它没有。

当我在没有"--gpu“模式选项的情况下运行训练过程时，它工作得很好，但速度很慢。(我也尝试过--gpu 0.8，0.4等。)

你知道怎么修复它吗？