深度学习程序，在GPU下运行和在CPU下运行，有何不同？

请问呢，如果你写了一个在CPU下运行的深度学习程序，想改成GPU下运行。 这个改动方便吗？

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

layjohns 发表于 2024-05-30 15:51
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

记得模型和数据都放到gpu上，如果一个放不下，要分布式，，，