[Help]: 请问如何实现指定GPU进行推理?
Closed this issue · 1 comments
leidingzhang commented
我使用python代码测试yolov8模型,虽然可以推理成功,但是查看代码未发现指定GPU的地方
laugh12321 commented
这里没有写(采用的是默认的Gpu 0),如果你有多个设备需要指定可以使用cuda.cudart.cudaSetDevice自己加上。
Closed this issue · 1 comments
我使用python代码测试yolov8模型,虽然可以推理成功,但是查看代码未发现指定GPU的地方
这里没有写(采用的是默认的Gpu 0),如果你有多个设备需要指定可以使用cuda.cudart.cudaSetDevice自己加上。