Is there any way to select the GPU ordinal for the tSNE execution?

CannyLab / tsne-cuda

GPU Accelerated t-SNE for CUDA with Python bindings

BSD 3-Clause "New" or "Revised" License

1.8k stars 129 forks source link

Is there any way to select the GPU ordinal for the tSNE execution? #128

Closed ruyian closed 1 year ago

ruyian commented 1 year ago

As per the title, I would like to use my 2nd GPU card for the t-SNE plot. Is there any way to achieve that? Shifting

DavidMChan commented 1 year ago

You should be able to use device=1 in the python command to change the selected device to the first ordinal. By default, if you have more than one device on the machine, I believe the latest version should try to use both devices.