cupy.fft.config.set_cufft_gpus#

cupy.fft.config.set_cufft_gpus(gpus)[source]#

设置多 GPU FFT 中使用的 GPU。

参数:: gpus (int 或 list of int) – 要使用的 GPU 数量或 GPU 列表。对于前一种情况，将使用前 gpus 个 GPU。

警告

此 API 目前处于实验阶段，未来版本中可能会有所更改。

另请参阅