下面的命令可以帮助我们查看Pytorch是否使用GPU: # 返回当前设备索引 torch.cuda.current_device() # 返回GPU的数量 torch.cuda.device_count() # 返回gpu名字,设备索引默认从0开始 torch.cuda.get_device_name(0) # cuda是否可用 torch.cuda.is_available() 我的结果是: 参考文档 pytorch中查看gpu信息怎么用 pytorch 查看 GPU 信息