在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中支持。