大佬教程收集整理的这篇文章主要介绍了我应该在哪里更改代码以告诉 PyTorch 不要使用 GPU?,大佬教程大佬觉得挺不错的,现在分享给大家,也给大家做个参考。
如How to tell PyTorch to not use the GPU?中所述,为了告诉 PyTorch 不使用 GPU,您应该更改 PyTorch 代码中的几行。 我应该在哪里进行更改? 需要修改的代码行在哪里? 我试图找到它但找不到...
在任何张量或 pytorch 模块上使用方法 .cpu()
会将组件传输到 cpu
,以便使用它进行计算。
另一个方向是使用方法.to("cpu")
。或者,您可以将“cpu”更改为“cuda”等其他设备的名称。
示例:
a)
@H_269_8@model = Mymodel().cpu() # move the model to the cpu x = data.cpu() # move the input to the cpu y = model(X)
b)
@H_269_8@model = Mymodel().to('cpu') # move the model to the cpu x = data.to('cpu') # move the input to the cpu y = model(X)
以上是大佬教程为你收集整理的我应该在哪里更改代码以告诉 PyTorch 不要使用 GPU?全部内容,希望文章能够帮你解决我应该在哪里更改代码以告诉 PyTorch 不要使用 GPU?所遇到的程序开发问题。
如果觉得大佬教程网站内容还不错,欢迎将大佬教程推荐给程序员好友。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。