下面的命令可以帮助我们查看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 信息