torch的torch device 有哪些
torch.device可以指定计算设备,常用的设备有:
-
CPU:torch.device('cpu'),用于在CPU上运行;
-
CUDA:torch.device('cuda'),用于在GPU上运行,需要安装CUDA并且GPU支持;
-
CUDA的某个设备:torch.device('cuda:0'),用于在指定的GPU上运行,0表示设备号,可以根据实际情况修改;
-
使用torch.device()时不指定任何参数,会自动选择可用的设备。
原文地址: https://www.cveoy.top/t/topic/clqA 著作权归作者所有。请勿转载和采集!