mne.cuda.set_cuda_device

mne.cuda.set_cuda_device(device_id, verbose=None)

Set the CUDA device temporarily for the current session.

Parameters:

device_idint

Numeric ID of the CUDA-capable device you want MNE-Python to use.

verbosebool | str | int | None

Control verbosity of the logging output. If None, use the default verbosity level. See the logging documentation and mne.verbose() for details. Should only be passed as a keyword argument.