mne.cuda.set_cuda_device

mne.cuda.set_cuda_device(device_id, verbose=None)[source]

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, or None

If not None, override default verbose level (see mne.verbose() and Logging documentation for more). If used, it should be passed as a keyword-argument only.