pytorch - directml 中查看设备支持情况

在windows 10/11中，如果显卡支持directx12, 可以用directml方式使用amd显卡进行加速（当然也支持n卡，不过n卡就老实用cuda吧） 

print(torch.dml.is_available()) #显示是否有dml设备

print(torch.dml.device_name(0)) ＃ 显示第０个ＤＭＬ设备名称

print(torch.dml.default_device()) ＃显示缺省ＤＭＬ设备ｉｄ

另外，pytorch支持的设备类型有：

cpu, cuda, ipu, xpu, mkldnn, opengl, opencl, ideep, hip, ve, fpga, ort, xla, lazy, vulkan, mps, meta, hpu

dml只在pytorhc-directml中支持。