out of memory with cunet

lltcggie / waifu2x-caffe

waifu2xのCaffe版

MIT License

8.07k stars 839 forks source link

Closed seiya-git closed 4 years ago

seiya-git commented 4 years ago

System: win 10 x64 Video card: Nvidia 1050 GTX

always when use cunet model, cui crashing with error: Check failed: error == cudaSuccess (2 vs. 0) out of memory

seiya-git commented 4 years ago

Fix: replace cudann_7.dll, use cudann for cuda 10.1 instead of 10.2