Closed leidingzhang closed 2 months ago
我使用python代码测试yolov8模型,虽然可以推理成功,但是查看代码未发现指定GPU的地方
这里没有写(采用的是默认的Gpu 0),如果你有多个设备需要指定可以使用cuda.cudart.cudaSetDevice自己加上。
我使用python代码测试yolov8模型,虽然可以推理成功,但是查看代码未发现指定GPU的地方