Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

在训练期间会同时使用CPU 和 GPU进行训练吗? #27

Closed
1154761334 opened this issue Dec 7, 2022 · 3 comments
Closed

在训练期间会同时使用CPU 和 GPU进行训练吗? #27

1154761334 opened this issue Dec 7, 2022 · 3 comments

Comments

@1154761334
Copy link

image
image

我在使用提供的代码时发现CPU 和GPU都busy,想咨询下是否同时使用GPU 和GPU一起训练,是为了最大限度利用资源吗?

@zezhishao
Copy link
Owner

训练过程仅仅在GPU上进行。STEP在预测阶段需要的计算量是较大的,不建议使用个人工作站级别的显卡做训练(3060Ti会比较慢)。我用的是两张V100或者两张3090。

@1154761334
Copy link
Author

老哥很感谢你的建议,目前我手头仅有的就是这块3060ti,本来计划想是买一块3090提升一下,但是考虑到最近价格上涨还是没买。转而从咸鱼上900一张买了两个p40。
1.打算通过两张p40先解决大显存有无问题。这样一些验证性的测试比如通过小的batch size 和 少量迭代数量可以先在我本地的环境上运行。
2.如果在少量迭代和 小的batch size下能有一个不错的效果,打算是租网上的显卡,也许几十块钱一个小时的云算力。如果后期能够做出一些成果也许能申请一台设备专门使用。
目前关于设备方面是这样一个计划。

@zezhishao
Copy link
Owner

好的。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants