Keras在AWS,TESLA V100上运行缓慢

时间:2018-08-16 14:50:44

标签: python amazon-web-services tensorflow keras cudnn

我正在使用带有一个Tesla V100 GPU的aws p3.2xlarge,而在我的个人数据集上训练模型时,与在笔记本电脑上训练相同模​​型时相比,我得到的性能较低。 这是性能指标(ETA)

TESLA V100
36672/332910 [==>...........................] - ETA: 1:56 - loss: 0.5794 - acc: 0.6664

GTX 950M
36480/299679 [==>...........................] - ETA: 30s - loss: 0.5894 - acc: 0.6652

当前的AWS设置:

  

操作系统:Ubuntu

     

CUDA版本:Cuda编译工具,版本9.1,V9.1.85

     

Cudnn版本:   7.2.1

     

Tensorflow版本:   1.10.0

0 个答案:

没有答案