코랩 GPU 관련 질문이 있습니다. (코드첨부)

Question

제 모델을 훈련시킬 때 에포크 당 시간이 너무 오래 걸려 코랩 GPU가 작동을 안하나?하고 알아본 결과 !nvidia-smi나 tf.device, device_lib.list_local_devices 등으로 tesla p100 GPU가 할당된 것은 확인했습니다.그러면 애초에 이 모델이 실행되는 시간이 오래 걸리는가 하는 질문이 있는데,층 3개(1층-뉴런300개, 2층-뉴런100개, 3층-뉴런10개)의 뉴럴 모델에 트레인 데이터셋을 넣어서 학습시키는 상황입니다.(트레인 데이터셋의 shape는 (55000, 28, 28)입니다. )이 때 55000개를 넣고 돌리는 하나의 에포크가 코랩에선 88초, 제 노트북에선 150초가 걸리네요..제가 보고 공부하는 책의 예제는 2초에 한 에포크를 돌리길래.. 저 코랩의 88초라는 시간이 코랩의 GPU가 돌아간 결과로 얻는 시간인 건지 궁금합니다.

코랩 GPU 관련 질문이 있습니다. (코드첨부)

0개의 답변

#Colab 목록