如果

torch.cuda.is_available()

的结果为真,但是代码运行后任务管理器的占用率低,把cuda加入列表,这样就能看到GPU已经在跑了。

但左侧显示仍然低,可以不管。

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐