Tensorflow - 리소스 소모 오류 CPU 및 GPU

Question

MNIST 데이터 세트로 CNN 코드를 테스트 할 때 GPU (GTX 1060, 8GB)를 사용하면 Resource Exhausted Error가 발생합니다.

그러나 동일한 코드를 테스트 할 때 CPU (i7-6700, RAM : 16GB)를 사용하면 리소스 소모 오류가 발생하지 않습니다.

이제 제 질문은 CPU & GPU의 차이점은 무엇입니까? 컴퓨팅 프로세스에는 뭔가 다른 점이 있습니까? 아니면 그냥 GPU 메모리 크기와 CPU RAM 크기의 차이 때문입니까? (사실, CPU 사용시 RAM 크기의 영향을 받습니까?)

Answer 1

GPU 메모리가 충분하지 않은 것일 수 있습니다. 귀하의 batchsize 줄이거 나 할 수있는 유일한 방법은 큰 GPU 메모리와 함께 더 나은 컴퓨터를 사용하는 것입니다.하지만 내 데이터의 한 batch_size는 컴퓨터가 계산하기에 충분히 큰 동일한 질문이 있습니다. 그래서 나는 그들 중 4 명에게 이미지를 나누어서 개별적으로 훈련 시켜서 최종 단계로 합쳤다. 결과는 나쁘지 않습니다.