admin管理员组

文章数量:1567915

当我们有多块显卡时,可以使用以下代码,明确指定用哪一块显卡来训练数据(cuda:0 还是 cuda:1)

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

本文标签: 显卡代码CUDA