指定模型在GPU上计算后怎么查看是不是在用GPU

当您指定模型在GPU上计算后，可以使用nvidia-smi命令来查看GPU的使用情况。该命令可以显示当前GPU的使用情况，包括GPU的使用率、内存使用情况等信息。

具体而言，可以在训练模型的同时在终端运行以下命令：

watch -n 1 nvidia-smi

该命令可以每秒钟刷新一次GPU的使用情况，让您实时监控GPU的使用情况。

如果您在运行模型时正确地将模型和数据移动到GPU上，并且模型的计算量足够大，您应该能够看到GPU使用率的上升，以及GPU内存的占用情况。如果GPU使用率不高，可能是因为模型的计算量不足或者数据集太小，无法充分利用GPU的计算能力。