在代码中的使用方法为: device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") model = Model() if torch.cuda.device_count() ...
確定! 回上一頁