search

ina-foss / inaSpeechSegmenter

CNN-based audio segmentation toolkit. Allows to detect speech, music, noise and speaker gender. Has been designed for large scale gender equality studies based on speech time per gender.
MIT License
717 stars 127 forks source link

Could I specify the GPU ?? #37

Closed ucas010 closed 4 years ago

ucas010 commented 4 years ago

hi,dear I want to specify the GPU for the seg? could you help me ? thx

DavidDoukhan commented 4 years ago

Dear @ucas010 , this is a tensorflow question. You need to set CUDA_VISIBLE_DEVICES environment variable. https://riptutorial.com/tensorflow/example/31877/use-a-particular-set-of-gpu-devices Kind regardss,