用不上GPU

[7W7W7W]

我在Windows上跑发现全程用CPU,然后换到Ubuntu上，还是CPU，环境是和文档一样的，这是为啥呀？

[HDL-YD]

作者代码gpu编号为7，你需要改成你使用的gpu的编号。一般从0开始是你的第一张显卡。