Skip to content

使用GPU训练需要修改哪些参数 #28

Description

@YukeeWang

感谢您的工作
我已经在电脑上成功运行了代码,但是貌似只在CPU上工作,我已经设置了gpu-options-allow-growth=True,且能看到有一部显存占用,但似乎训练都是再CPU上完成的,我需要调整别的什么配置么?多谢

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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