How to specify GPU index?

serizba / cppflow

Run TensorFlow models in C++ without installation and without Bazel

https://serizba.github.io/cppflow/

MIT License

779 stars 177 forks source link

Open xfchangwei opened 2 years ago

xfchangwei commented 2 years ago

How to specify GPU index? For example, I have multiple GPU devices

serizba commented 1 year ago

Hi @xfchangwei

I guess this can be easily done with an environment variable:

export CUDA_VISIBLE_DEVICES=0