PyTorch 设备选择: 'cuda:0' vs. 'cpu'

这行代码用于判断当前环境是否支持 GPU，并根据结果选择合适的计算设备。

• 如果当前环境支持 GPU，则将 device 设置为 'cuda:0'，表示使用第一个可用的 GPU 进行计算。
• 否则，将 device 设置为 'cpu'，表示使用 CPU 进行计算。

这种方法的优势在于，它可以让代码在不同的环境下运行，并自动选择最佳的设备进行计算。例如，在具有 GPU 的机器上，代码将使用 GPU 进行训练，从而获得更快的速度；而在没有 GPU 的机器上，代码将使用 CPU 进行训练，仍然可以正常运行。