Add option to select GPU.

[marl]

Add option to select GPU.

[jongwook]

Hi, thanks for the suggestion. While I could add the CLI argument in the future, you can manually select the GPU using an environment variable:

CUDA_VISIBLE_DEVICES=2 crepe audio.wav

where 2 is the GPU id. You may also force it to run on CPU by giving -1.

[xcyoloxcy]

But when CREPE is combined in some other code, the GPU cannot be changed by modify the environment variable. How to empty the memory CREPE occupied, or run other code in another GPU?

[seyong92]

Thanks for your help!