ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

torch.device()基础用法

2022-05-20 22:05:42  阅读:35  来源: 互联网

标签:torch 用法 cuda device GPU model data


用途:用于指定将数据或者模型放在GPU/CPU

#1.通常用法
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
data = data.to(device)
model = model.to(device)
'''
    1.先创建device对象
    2.to函数指定数据或者模型放到哪里
'''

#2.将构建的tensor或者模型放到指定设备上(GPU)
torch.device('cuda',0) #这里的0指的是设备的序号
torch.device('cuda:0')

 

 

标签:torch,用法,cuda,device,GPU,model,data
来源: https://www.cnblogs.com/jiu-fang/p/16293816.html

专注分享技术,共同学习,共同进步。侵权联系[admin#icode9.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有