-
在进行深度学习训练时,下载的代码有的需要GPU计算,而电脑又没有GPU,此时可以在代码开头添加如下代码获取当前计算机是否有GPU
# gpu or not
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
然后在代码中 用 .to(device)替换.cuda() 或者.cpu() 即可在有GPU的计算机上通过GPU计算加速,在没有GPU的计算机上用CPU计算,不过速度较慢
-
在写深度学习算法时,也尽可能采用这种同样的代码,方便在任何机器上运行。