Linux如何查看CUP及其使用情况 一、指定使用哪个GPU 1.先下载gpustat: pip install gpustat 2.watch -n 2 --color gpustat --c 即可查看这里可以看到4是可用的,然后我们再用DEVICE = torch.device(“cuda:4”)设置需要使用的GPU 二、将计算放到GPU中进行 定义好网络之后要将训练、测试数据还有网络本身都都到在GPU上,如: 测试: 训练: 网络: 这样在之后的训练过程中就是在GPU上了