[Help]: 请问如何实现指定GPU进行推理？

[leidingzhang]

我使用python代码测试yolov8模型，虽然可以推理成功，但是查看代码未发现指定GPU的地方

[laugh12321]

这里没有写(采用的是默认的Gpu 0)，如果你有多个设备需要指定可以使用cuda.cudart.cudaSetDevice自己加上。