在pytorch中用一句话：有cuda就用cuda，没cuda就用cpuDEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu') 即可版权声明：本文为zh515858237原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。原文链接：https://blog.csdn.net/zh515858237/article/details/124151431