memory capacity error while training using GPU
sahanHe opened this issue · 0 comments
sahanHe commented
I tried to train the model for DUTS dataset using a GPU and i get the following error message
My dependencies are
Cuda 10.0.130
cudann 7.6.4
tensorflow-gpu=1.15.2
opencv=3.4.5.20
my GPU is RTX2080Ti
thank you