Sélectionner un appareil d’exécution

Avec les API Windows ML, vous pouvez sélectionner un appareil sur lequel vous préférez évaluer votre modèle. Vous sélectionnez un appareil lorsque vous créez une session. Pour ce faire, vous allez utiliser la classe LearningModelDevice, qui définit l’appareil utilisé pour évaluer le modèle Machine Learning.

Ici, nous allons apporter des modifications dans le classifier.cs fichier de l’application précédente, afin de pouvoir exécuter notre évaluation de modèle sur le GPU de nos appareils. Nous n’avons pas touché ce fichier dans le tutoriel précédent, car il a été généré automatiquement par le générateur de code WinML.

1. Ouvrez le classifier.cs fichier et ajoutez le code suivant à la CreateFromStreamAsync méthode.

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

Pour choisir un autre appareil d’exécution, basculez simplement le champ DirectX vers un autre.

2. À présent, vous pouvez sélectionner cet appareil lorsque vous créez une session. Modifiez le LearningModelSession constructeur pour spécifier l’appareil d’exécution.

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

La méthode CreateFromStreamAsync se présente désormais comme suit :

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;
        }

C’est tout! Vous avez sélectionné avec succès le GPU pour l’évaluation du modèle.

Si vous ne spécifiez pas d’appareil, le système utilise la valeur par défaut. Nous vous recommandons d’utiliser Default pour obtenir la flexibilité permettant au système de choisir pour vous à l’avenir.

Windows AI prend en charge l’utilisation d’un appareil que l’appelant a déjà créé. Pour en savoir plus sur la création avancée d’appareils, consultez la documentation Créer une session.