مشکل در اجرای تنسورفلو با gpu

Question

سلام. من می خوام کدم رو با gpu اجرا کنم (تنسورفلو) ولی توی ترینینگ زمانی که هنوز هیچ epoch نرفته ارور زیر رو میده:
OOM when allocating tensor with shape...
توی نت چیز به درد بخوری پیدا نکردم.
واقعا gpu این همه دردسر داره؟!!! مدل gpu من gtx1070 هستش و شبکم conv-max pool - conv-maxpool-FC-softmax هست
ممنون

Answer 1

سلام وقت بخیر
ن اتفاقا سختی خاصی ندارد.
همه چیز بستگی به منابع شما دارد:::

لینک زیر را بررسی کنید.
http://stackoverflow.com/questions/39076388/tensorflow-deep-mnist-resource-exhausted-oom-when-allocating-tensor-with-shape

Answer 2

حل شد مشکل:
توی sess.run زمانی که می خواستم feed_dict کنم کل تست و ترین رو برای محاسبه دقت داده بودم که به صورت batch اعمال کردم و حل شده. فعلا البته!