search

ceccocats / tkDNN

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms
GNU General Public License v2.0
720 stars 209 forks source link

where to set cuda device? #261

Closed sunian6183 closed 3 years ago

sunian6183 commented 3 years ago

where to set cude device for choosing which one GPU to run ?

mive93 commented 3 years ago

Do you mean when you have more than one GPU? It's the same as other programs

export CUDA_VISIBLE_DEVICES=1

or

CUDA_VISIBLE_DEVICES=1 ./demo <all the param>
mive93 commented 3 years ago

Closing for inactivity