GTX1060 CUDA out of memory

WangYueFt / dcp

356 stars 90 forks source link

GTX1060 CUDA out of memory #18

Closed tuduku closed 4 years ago

tuduku commented 4 years ago

thank you for your work i am wondering that what GPU were you using in trainning and testing

i'm using GTX1060 python3.6 PyTorch1.0 Cuda9.0 but i get "RuntimeError: CUDA out of memory." with batch_size=8 while trainning and get the same error with test_batch_size=4 while testing

tuduku commented 4 years ago

i'm sorry i didn't notice the first issue