Skip to content

CUDA可用,但是推理时还是用的CPU #341

Open
@sxsp

Description

@sxsp

代码如下
7ae6bec1569e16699816e084e7d462d
Loading checkpoint shards时会报下面的warning
image
可用推理,但是似乎用的是cpu
但我的cuda显示gpu是可用的(如下
image
为什么呀

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions