概述
tensor数据的cuda方法返回变量值的device为cuda，并不会直接移动当前变量到GPU。
举例：
B = A.cuda()
其中A为CPU变量，那么执行上面语句后，A依旧在CPU上，创建的新的数据B是A在GPU上面的拷贝，当然单独执行A.cuda()，A也依旧在CPU上面。
import torch
A = torch.empty(4,1,2,2)
print(A.device)
print(A)
print(A.cuda().device)
print(A.cuda())
# A.cuda()
A = A.cuda()
print(A.device)
# A = A.to('cuda:0')
最后
以上就是单薄铃铛为你收集整理的pytorch 数据放到GPU上面的全部内容，希望文章能够帮你解决pytorch 数据放到GPU上面所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复