ONNX runtime指定GPU
资料参考：链接self.onnx_session = onnxruntime.InferenceSession(onnx_path)self.onnx_session.set_providers(['CUDAExecutionProvider'], [ {'device_id': 1}])在onnx_session.set_providers中设置即可
·
资料参考：链接
self.onnx_session = onnxruntime.InferenceSession(onnx_path)
self.onnx_session.set_providers(['CUDAExecutionProvider'], [ {'device_id': 1}])
在onnx_session.set_providers中设置即可
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
4
0- 0
已为社区贡献2条内容
所有评论(0)