Pilih perangkat eksekusi

Dengan API Windows ML, Anda dapat memilih perangkat yang lebih Anda sukai untuk mengevaluasi model Anda. Anda memilih perangkat saat membuat sesi. Untuk melakukannya, Anda akan menggunakan kelas LearningModelDevice, yang menentukan perangkat yang digunakan untuk mengevaluasi model pembelajaran mesin.

Di sini, kita akan membuat perubahan dalam classifier.cs file aplikasi sebelumnya, sehingga kita dapat menjalankan evaluasi model di GPU perangkat kita. Kami tidak menyentuh file itu di tutorial sebelumnya, karena secara otomatis dihasilkan oleh Generator Kode WinML.

1. Buka file classifier.cs dan tambahkan kode berikut ke metode CreateFromStreamAsync.

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

Untuk memilih perangkat eksekusi yang berbeda, cukup alihkan bidang DirectX ke yang berbeda.

2. Sekarang, Anda dapat memilih perangkat ini saat membuat sesi. LearningModelSession Ubah konstruktor untuk menentukan perangkat eksekusi.

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

Metode CreateFromStreamAsync sekarang sebagai berikut:

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

Itu saja! Anda berhasil memilih GPU untuk evaluasi model.

Jika Anda tidak menentukan perangkat, sistem akan menggunakan Default. Sebaiknya gunakan Default untuk mendapatkan fleksibilitas memungkinkan sistem memilih untuk Anda di masa mendatang.

Windows AI mendukung penggunaan perangkat yang telah dibuat pemanggil. Untuk mempelajari selengkapnya tentang pembuatan perangkat tingkat lanjut, tinjau dokumentasi Buat sesi.