Running without CUDA (CPU only)

[martiningram]

Hi,

I am just wondering if there is a way of running this model on a machine without a GPU? Running make attempts to find CUDA.

Thanks!

[endernewton]

you can install cuda without gpu. haven't tried that though.

[jwnsu]

Current codes run fine on CPU, need to change NMS mode to CPU only.

…

On Tue, Mar 28, 2017 at 1:21 PM, Xinlei Chen ***@***.***> wrote: you can install cuda without gpu. haven't tried that though. — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#25 (comment)>, or mute the thread <https://github.com/notifications/unsubscribe-auth/AAM4K-eGWG5K1k8TuAhbGaKNmLFFny0Uks5rqWu7gaJpZM4Mkvv3> .

[endernewton]

Ahh I see. thanks for the update. Closing.

[guotong1988]

@jwnsu Could you please describe how to run it on CPU in detail? Thank you.

[guotong1988]

Finally it shows:
iter: 20 / 70000, total loss: 2.201489

You need to:
change nms_wrapper.py.
change setup.py.
make at ./lib.
put the data at tf-faster-rcnn-master/data/VOCdevkit2007/VOC2007
and here are the parameters:

  parser.add_argument('--cfg', dest='cfg_file',
                      help='optional config file',
                      default="/home/gt/tf-faster-rcnn-master/experiments/cfgs/vgg16.yml", type=str)
  parser.add_argument('--weight', dest='weight',default="/home/gt/tf-faster-rcnn-master/vgg16.ckpt",
                      help='initialize with pretrained model weights',
                      type=str)
  parser.add_argument('--imdb', dest='imdb_name',
                      help='dataset to train on',
                      default='voc_2007_trainval', type=str)
  parser.add_argument('--imdbval', dest='imdbval_name',
                      help='dataset to validate on',
                      default='voc_2007_trainval', type=str)
  parser.add_argument('--iters', dest='max_iters',
                      help='number of iterations to train',
                      default=70000, type=int)
  parser.add_argument('--net', dest='net',
                      help='vgg16 or res101',
                      default='vgg16', type=str)

[bhack]

@guotong1988 Can you open a PR with a CPU option.. could be useful for users with GPU memory limits..

[caunion]

I think you can add a parameter in commandline to force CPU. i.e.

./experiments/scripts/train_faster_rcnn.sh 1 pascal_voc vgg16 USE_GPU_NMS False

[Dinkz]

@guotong1988
Can I know what change you did inside setup.py ?

[luoru]

@guotong1988
what change ocured in nms_wrapper.py & setup.py?

[dancsalo]

@caunion's comment contains all that you need to do to force CPU. No need to change nms_wrapper.py or setup.py. Just tried it myself!

[joeliven]

@endernewton great implementation of Faster RCNN. Really appreciate it.
Question though: I can't get this to work on CPU only. I've read through the above and tried those suggestions but can't seem to get it to work. @caunion @guotong1988 has anybody had success in making this work on CPU? If so, how?
Thanks in advance for any suggestions. Much appreciated!

[endernewton]

So what is the error msg?

…

Sent from my iPhone

On Jun 5, 2017, at 16:27, Joel Iventosch ***@***.***> wrote: @endernewton great implementation of Faster RCNN. Really appreciate it. Question though: I can't get this to work on CPU only. I've read through the above and tried those suggestions but can't seem to get it to work. @caunion @guotong1988 has anybody had success in making this work on CPU? If so, how? Thanks in advance for any suggestions. Much appreciated! — You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or mute the thread.

[riadhayachi]

use this command
./tools/demo.py cpu