search

microsoft / DirectML

DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.
MIT License
2.24k stars 299 forks source link

The python crashed #653

Open AAxjcy opened 1 month ago

AAxjcy commented 1 month ago

CPU: i7-12700H GPU: Intel(R) Iris(R) Xe Graphics\x00 Open Python and enter the following code:

import torch
import torch_directml 
dml=torch_directml.device(1) #here dml is Intel(R) Iris(R) Xe Graphics\x00
x=torch.tensor([0],dtype=torch.float32).to(dml)
A=torch.nn.Linear(1,1).to(dml)
A(x)

Then my python crashes steadily without errors