RuntimeError: CUDA error: invalid device ordinal
作者:XD / 发表: 2022年5月24日 02:42 / 更新: 2022年7月27日 00:39 / 编程笔记 / 阅读量:2136
RuntimeError: CUDA error: invalid device ordinal
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") # update "cuda:0" number