分类：Pytorch切换cpu和gpu

2021
03-02

Pytorch如何切换 cpu和gpu的使用详解

前言，在pytorch中，当服务器上的gpu被占用时，很多时候我们想先用cpu调试下代码，那么就需要进行gpu和cpu的切换。方法1：x.to(device)把device作为一个可变参数，推荐使用argparse进行加载：使用gpu时：device='cuda'x.to(device)#x是一个tensor，传到cuda上去使用cpu时：device='cpu'x.to(device)方法2：使用x.cuda()+CUDA_VISIBLE_DEVICES很多贴子中说，使用x.cuda()和x.to('cuda')虽然是等效的，但是x.cuda()的缺点是... 继续阅读 >