网站首页 pytorh
-
pytorch中.to(device) 和.cuda()的区别说明
原理.to(device)可以指定CPU或者GPUdevice = torch.device("cuda:0" if torch.cuda.is_available()&nbs...
2021-07-16 longtao100 python 1216 ℃ 0 评论
- 开源分类
- 开源网标签
原理.to(device)可以指定CPU或者GPUdevice = torch.device("cuda:0" if torch.cuda.is_available()&nbs...
2021-07-16 longtao100 python 1216 ℃ 0 评论