与 colaboratory 相比,Python 脚本在 google 计算实例上运行速度较慢

与 24-vCPU 谷歌计算实例相比,为什么我的机器学习 python/tensorflow 脚本在 colab 上运行得更快?

在 colab 上调用: !/content/myscript.py

谷歌计算实例上的调用: !/home/prj1/myscript.py

colab 上的纪元时间:0.8 秒

谷歌计算实例的纪元时间:2.0 秒

在这两种情况下,我都使用 tensorflow 1.11 和 python 2.7,myscript.py 是唯一运行的程序,并且没有使用 GPU。该脚本每 10 个时期在屏幕上显示训练进度(纯文本,无图形)。


偶然的你
浏览 204回答 2
2回答

慕容3067478

通过遵循提高 CPU 上流行深度学习框架性能的技巧中推荐的优化,我能够将 epoch 时间降低到 1.1 秒这是代码:import osN_CORES                       = int(os.cpu_count()/2)os.environ["OMP_NUM_THREADS"] = str(N_CORES)os.environ["KMP_BLOCKTIME"]   = "30"os.environ["KMP_SETTINGS"]    = "1"os.environ["KMP_AFFINITY"]    = "granularity=fine,verbose,compact,1,0"config_sess                   = tf.ConfigProto(intra_op_parallelism_threads=N_CORES, inter_op_parallelism_threads=2, allow_soft_placement=True, device_count = {'CPU': N_CORES})with tf.Session(config=config_sess) as sess:
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python