import onnxruntime as ort
print(ort.get_available_providers())

如果输出List里面有

CUDAExecutionProvider则表示可用GPU,如果没有只能使用CPU了
Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐