pytorch的代码 CPU改为GPU_paddle cpu版本的代码如何置换gpu版本

作者：黑客灵魂 | 2024-07-19 12:13:33

踩

paddle cpu版本的代码如何置换gpu版本

pytorch的代码 CPU改为GPU
找一下几个地方改动

一、网络模型

mymodule = MyModule()
mymodule = mymodule.cuda()
1
2

二、数据（输入、标记）

        images,targets = data
        images = images.cuda()
        targets = targets.cuda() #训练数据和测试数据集都要进行此操作
1
2
3

三、损失函数

# 损失函数
loss = nn.CrossEntropyLoss()
loss = loss.cuda()
1
2
3

上面的写法，在无GPU时会报错。更好的写法是有GPU用GPU，无GPU用CPU

mymodule = MyModule()
if torch.cuda.is_available():
    mymodule = mymodule.cuda()
1
2
3

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/黑客灵魂/article/detail/851434