cupy.fft.config.set_cufft_gpus#
- cupy.fft.config.set_cufft_gpus(gpus)[source]#
Set the GPUs to be used in multi-GPU FFT.
- Parameters
gpus (int or list of int) – The number of GPUs or a list of GPUs to be used. For the former case, the first
gpus
GPUs will be used.
Warning
This API is currently experimental and may be changed in the future version.
See also