在pytorch中,即使是有GPU的机器,它也不会自动使用GPU,而是需要在程序中显示指定。调用model.cuda(),可以将模型加载到GPU上去。这种方法不被提倡,而建议使用model.to(device)的方式,这样可以显示指定需要使用的计算资源,特别是有多个GPU的情况下。
pytorch中 model.cuba的作用是什么?
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
pytorch 中的梯度计算
永远的小白虾
2020-07-07 15:03:38
Pytorch批量数据读取方法详解: DataLoader
miguemath
2020-07-08 04:10:07
subprocess.py报错:FileNotError: [Errno 2] No such file or directory: java: java
清晨的光明
2020-07-08 02:37:26
Pytorch知识点与问题解决
ME_Seraph
2020-07-08 00:56:51
Pytorch: dataloader的一些使用心得
WinterShiver
2020-07-07 23:40:48
Pytorch|YOWO原理及代码详解(二)
NotFound1911
2020-07-07 23:39:48
Pytorch|YOWO原理及代码详解(三)
NotFound1911
2020-07-07 23:39:48
Pytorch|YOWO原理及代码详解(一)
NotFound1911
2020-07-07 23:39:37
Pytorch显存问题
Peanut_范
2020-07-07 17:46:09
Kaggle手写数字识别(Digit Recognizer)记录
不跑步就等肥
2020-07-07 15:12:31
Pytorch搭建神经网络基本步骤(文末附pytorch实现AlexNet)
不跑步就等肥
2020-07-07 15:12:31
强化学习--Pytorch--DQN扩展以及Policy Gradient网络结构
Chasing中的小强
2020-07-07 11:10:39
强化学习--Pytorch--DDPG
Chasing中的小强
2020-07-07 11:10:37
强化学习--Pytorch--DQN
Chasing中的小强
2020-07-07 11:10:37