在windows 10/11中,如果显卡支持directx12, 可以用directml方式使用amd显卡进行加速(当然也支持n卡,不过n卡就老实用cuda吧)
print(torch.dml.is_available()) #显示是否有dml设备
print(torch.dml.device_name(0)) # 显示第0个DML设备名称
print(torch.dml.default_device()) #显示缺省DML设备id
另外,pytorch支持的设备类型有:
cpu, cuda, ipu, xpu, mkldnn, opengl, opencl, ideep, hip, ve, fpga, ort, xla, lazy, vulkan, mps, meta, hpu
dml只在pytorhc-directml中支持。