分类：pytorch网络

2021
07-30

将pytorch的网络等转移到cuda

神经网络一般用GPU来跑，我们的神经网络框架一般也都安装的GPU版本，本文就简单记录一下GPU使用的编写。GPU的设置不在model，而是在Train的初始化上。第一步是查看是否可以使用GPUself.GPU_IN_USE=torch.cuda.is_available()就是返回这个可不可以用GPU的函数，当你的pytorch是cpu版本的时候，他就会返回False。然后是：self.device=torch.device('cuda'ifself.GPU_IN_USEelse'cpu')torch.device是代表将torch.tensor分配到... 继续阅读 >