方式一
tudui = Tudui()
if torch.cuda.is_available():
tudui = tudui.cuda (网络调用cuda后要再返回给网络)
………………………………………………………………………………
if torch.cuda.is_available():
loss = loss.cuda()
………………………………………………………………………………
if torch.cuda.is_available():
imgs = imgs.cuda()
targets = targets.cuda()
方式二
放开头
device = torch.device(‘cuda’)
………………………………………………………………………………
‘tudui = Tudui()
tudui = tudui.to(device)
………………………………………………………………………………
loss = loss.to(device)
………………………………………………………………………………
images = images.to(device)
targets = targets.to(device)
Original: https://blog.csdn.net/xiao_bai_bu_hei/article/details/122496915
Author: xiao_bai_bu_hei
Title: 调用cuda在GPU上运行,能调用CUDA的地方
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/511524/
转载文章受原作者版权保护。转载请注明原作者出处!