search

PRBonn / bonnet

Bonnet: An Open-Source Training and Deployment Framework for Semantic Segmentation in Robotics.
GNU General Public License v3.0
323 stars 89 forks source link

Using /gpu:1 #41

Closed thanhngotud closed 5 years ago

thanhngotud commented 5 years ago

Hi, I'm currently building bonnet application in PX2, but I'm not using ROS. At the moment, I can install the bonnet. But when I run bonnet, there is the problem of GPU device. In PX2, we have 2 gpu: iGPU and dGPU. The iGPU is the one only support fp16 and its locate in /gpu:1. I just want to ask is there any way to make Bonnet using /gpu:1, because at the moment it only accpet for gpu:0.

Best regards Thanh

tano297 commented 5 years ago

Hello,

Are you using tensorflow or tensorRT?

thanhngotud commented 5 years ago

I’m using tensorRT

daniel89710 commented 5 years ago

"export cuda_visible_devices=1" enables us to use iGPU on PX2.

thanhngotud commented 5 years ago

Thanks for your help