search

charlesq34 / pointnet

PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
Other
4.73k stars 1.45k forks source link

GPU to use #68

Closed yoshiyama closed 6 years ago

yoshiyama commented 6 years ago

I would like to use GPU for processing faster. Therefore I would like to know how to set GPU. Optional arguments include GPU setting ( --gpu GPU GPU to use [default: GPU 0]). What does "0" in GPU 0 here mean ? For example it does not process faster with --gpu 0 and --gpu 1, respectively.

My PC has a NVIDIA GeForde 1080.

Let me know how to use GPU.

themmes commented 6 years ago

If your computer/server has a (or multiple) dedicated GPU(s) these are generally numbered starting with 0. If Tensorflow is not able to find a GPU it will log this explicitly.