文件名称:cpu_gpu_test
文件大小:21.81MB
文件格式:ZIP
更新时间:2023-04-15 07:58:46
tensorflow pytorch minist gpu cpu
提供tensorflow和pytorch在gpu上训练minist数据集的精简样例源码,包括对比相同运算在gpu和cpu上的时间差异
【文件预览】:
cgpu
----tensorflow_minist.py(617B)
----pytorch_minist.py(6KB)
----MNIST()
--------raw()
--------processed()
----validate_cgpu.py(262B)