Selección de un dispositivo de ejecución

Con las API de Windows ML, puede seleccionar un dispositivo en el que prefiera evaluar el modelo. Al crear una sesión, seleccione un dispositivo. Para ello, usará la clase LearningModelDevice, que define el dispositivo que se usa para evaluar el modelo de Machine Learning.

Aquí, realizaremos cambios en el archivo de la classifier.cs aplicación anterior, por lo que podemos ejecutar la evaluación del modelo en la GPU de nuestros dispositivos. No hemos tocado ese archivo en el tutorial anterior, ya que lo generó automáticamente el generador de código WinML.

1. Abra el classifier.cs archivo y agregue el código siguiente al CreateFromStreamAsync método .

// Select GPU or another DirectX device to evaluate the model.
LearningModelDevice device = new LearningModelDevice(LearningModelDeviceKind.DirectX);

Para elegir un dispositivo de ejecución diferente, simplemente cambie el campo DirectX a otro diferente.

2. Ahora, puede seleccionar este dispositivo al crear una sesión. Cambie el LearningModelSession constructor para especificar el dispositivo de ejecución.

// Create the evaluation session with the model and device.
learningModel.session = new LearningModelSession(learningModel.model, device);

El método CreateFromStreamAsync ahora es el siguiente:

CreateFromStreamAsync(IRandomAccessStreamReference stream)
        {
            classifierModel learningModel = new classifierModel();
            learningModel.model = await LearningModel.LoadFromStreamAsync(stream);

            // Select GPU or another DirectX device to evaluate the model.
            LearningModelDevice device = new LearningModelDevice(LearningModelDeviceKind.DirectX);
            
            // Create the evaluation session with the model and device.
            learningModel.session = new LearningModelSession(learningModel.model, device);
            learningModel.binding = new LearningModelBinding(learningModel.session);
            return learningModel;
        }

¡Es todo! Ha seleccionado correctamente la GPU para la evaluación del modelo.

Si no especifica un dispositivo, el sistema usa Default. Se recomienda usar Default para obtener la flexibilidad de permitir que el sistema lo elija en el futuro.

Windows AI admite el uso de un dispositivo que el autor de la llamada ya ha creado. Para más información sobre la creación avanzada de dispositivos, revise la documentación Creación de una sesión.

Nota:

Para obtener información sobre cómo seleccionar el dispositivo de ejecución, revise la documentación de LearningModelDevice.