search

chenyuntc / pytorch-book

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)
MIT License
11.99k stars 3.73k forks source link

关于CPU与GPU的问题，请教大家 #192

Open ZaynStark opened 4 years ago

ZaynStark commented 4 years ago

我的电脑只有i7 9700的核显，没有配置独立显卡，能否在pytorch框架下，只使用CPU训练书中的模型，不是很清楚，想请教大家

Canadalynx commented 4 years ago

只要在跑的时候占用的计算资源(模型加上batch的数据大小等)不超过cpu的能用内存就能跑。就是速度明显有骑驴和开奔驰的区别。

iydon commented 4 years ago

仅供参考: 完全使用第九章例子, CPU 为 Intel Core i5-7200U @ 4x 3.1GHz, 训练时间大约为 7 小时; 后来借了一台机器, CPU 为 Intel Xeon E3-1225 v5 @ 4x 3.7GHz, GPU 为 Quadro K2200, 不到一节课 (30分钟) 不到就训练完了. 虽然没有控制变量, 但可以直观感受, 如果没有 GPU, 单纯使用 CPU, 效果不是很好, 验证自己思路的实验次数也会相应变少.