对于torch.device()
函数,参数可以是字符串,指定要使用的设备(如"cpu"
或"cuda"
),也可以是一个整数,表示要使用的GPU设备的索引号。例如,torch.device("cuda:0")
表示使用第一个可用的GPU。 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
torch.device()
最新推荐文章于 2024-03-28 07:42:00 发布
对于torch.device()
函数,参数可以是字符串,指定要使用的设备(如"cpu"
或"cuda"
),也可以是一个整数,表示要使用的GPU设备的索引号。例如,torch.device("cuda:0")
表示使用第一个可用的GPU。 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")