욜로 - 훈련 내 GPU를 사용하여 모델 욜로 감지 기능을 CPU에서 작동하지만 GPU

Question

에 내가 사용한 적이 없습니다 tensorflow - 엔비디아 1060 3GB, 모든 것이 괜찮 았는데. PARAM 1.0 --gpu와

지금 나는 내 자신의 모델을 생성하는 것을 시도하고있다. 내가 읽은에서 그가 통신을 시작할 수 Tensorflow 내 GPU를 볼 수 있습니다 "이름 : 지포스 GTX를 1060 주요 6 부 : 1 memoryClockRate (GHz의) : 1.6705" "totalMemory : 3.00GiB freeMemory에서 : 2.43GiB"

프로그램 데이터를로드, 나는 오류 다음있어 학습을 시작하려고 할 때 어쨌든, 나중에 : 그것을 사용하려고하면 내가 확인했습니다

: "CUDA_ERROR_OUT_OF_MEMORY 장치에서 832.51M (872,952,320 바이트)를 할당하는 데 실패했습니다" 내 다른 GPU (인텔 630),하지만 그렇지 않습니다. 내가 "--gpu"모드 옵션을 사용하지 않고 기차 프로세스를 실행으로

, 그것은하지만 천천히, 잘 작동합니다. (나는 또한 시도 0.8, 0.4 요법. --gpu 한)

어떤 생각이 어떻게 그것을 해결하기 위해?

Answer 1

과제를 해결한다. 구성 파일에서 배치 크기와 이미지 크기를 변경하면 올바르게로드되지 않았으므로 도움이되지 않는 것 같습니다. 내 GPU가 단계를 계산할 수 있도록 defaults.py 파일로 이동하여 더 낮은 곳으로 변경해야했습니다.