我正在使用带有一个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