P30.利用GPU训练(一)

tudui = tudui.cuda()
loss_fn = loss_fn.cuda()
imgs = imgs.cuda()
targets = targets.cuda()
if torch.cuda.is_available():
本文介绍了如何在使用CUDA的GPU上进行深度学习模型的高效训练,包括cuda设备初始化、损失函数和数据集的GPU转移,适用于PyTorch开发者。

tudui = tudui.cuda()
loss_fn = loss_fn.cuda()
imgs = imgs.cuda()
targets = targets.cuda()
if torch.cuda.is_available():
976
364
4238

被折叠的 条评论
为什么被折叠?