您好,欢迎来到欧得旅游网。
搜索
您的当前位置:首页⽤方法 to() 可以将 Tensor 在CPU和GPU(需要硬件支持)之间相互移动

⽤方法 to() 可以将 Tensor 在CPU和GPU(需要硬件支持)之间相互移动

来源:欧得旅游网
# 以下代码只有在PyTorch GPU版本上才会执⾏行行
if torch.cuda.is_available():
device = torch.device("cuda") # GPU
y = torch.ones_like(x, device=device) # 直接创建一个在GPU上的Tensor
x = x.to(device) # 等价于 .to("cuda")
z = x + y
print(z)
print(z.to("cpu", torch.double)) # to()还可以同时更改数据类型

 

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- ovod.cn 版权所有 湘ICP备2023023988号-4

违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务