cpu与gpu的切换_cupy.asarray如何从cpu移到gpu

作者：酷酷是懒虫 | 2024-07-19 12:19:06

踩

cupy.asarray如何从cpu移到gpu

方法：
在训练之前加入这一句

#选择使用gpu，如果没有gpu就使用cpu
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
1
2
然后，在训练的代码中会找到 .cuda() ，将所有的都替换成为.to(device) 即可
（在网络后边和输入输出后边都需要改）

即将

model=model.cuda()
x=x.cuda()
y=y.cuda()
1
2
3
上边的改为下边的！！！！！！！

model=model.to(device)
x=x.to(device)
y=y.to(device)

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