cppflow icon indicating copy to clipboard operation
cppflow copied to clipboard

How to specify GPU index?

Open xfchangwei opened this issue 2 years ago • 2 comments

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

xfchangwei avatar May 07 '22 06:05 xfchangwei

Hi @xfchangwei

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

export CUDA_VISIBLE_DEVICES=0

serizba avatar Sep 23 '22 11:09 serizba