|
|
一、获取测试代码
最新的benchmark代码有bug,使用稍旧一点的版本就可以，代码地址：https://github.com/tensorflow/be ... _tf_v1.5_compatible
二、测试脚本
1）GPU
python tf_cnn_benchmarks.py --local_parameter_device=gpu --num_gpus=1 --variable_update=replicated --batch_size=64 --model=vgg16
2）CPU
python tf_cnn_benchmarks.py --device=CPU --variable_update=parameter_server --num_intra_threads=0 --batch_size=64 --model=vgg16 --data_format=NHWC
|
|
发表于 2019-2-1 10:06:24
