search

scaelles / OSVOS-TensorFlow

One-Shot Video Object Segmentation
http://vision.ee.ethz.ch/~cvlsegmentation/osvos/
GNU General Public License v3.0
438 stars 132 forks source link

What are the GPU memory requirements to run this program? #12

Closed kaisark closed 6 years ago

kaisark commented 6 years ago

What are the GPU memory requirements to run this program?

Can this code run on a Nvidia TX1 with 4GB of memory?

Will the code (if configured) run using multiple GPU if I push to AWS P2 instance?

kmaninis commented 6 years ago

Hello,

We checked on TensorFlow 1.1, and training takes approximately 6.4G of GPU memory. There may be optimizations for the newest TF version, which we did not try.

I hope it helps.