pytorch中如何查看模型或者数据是否使用gpu
https://blog.csdn.net/jzwong/article/details/110955851

如何使用linux查看cpu的使用
vmstat
top
dstat
https://cloud.tencent.com/developer/article/1478176

使用pytorch使用gpu和cpu时，要注意数据的位置在哪里？