提供tensorflow和pytorch在gpu上训练minist数据集的精简样例源码,包括对比相同运算在gpu和cpu上的时间差异