Can the project run on GPU?

SaoYan / bgsCNN

Background subtraction using deep learning method.

GNU General Public License v3.0

191 stars 61 forks source link

Closed Wisgon closed 6 years ago

Wisgon commented 6 years ago

I use tensorflow-gpu to run the code, but it still very slow and the gpu-util is always 0, and the memory of my two GPUs both full.

Wisgon commented 6 years ago

@SaoYan

SaoYan commented 6 years ago

I ran the code on GPU.

It is true that this code is memory consuming. More improvements are encouraged.

Wisgon commented 6 years ago

OK, thank you for replying.