tensorflowgpu没有加速训练

Question

我最近建立了一个CNN，然后我安装了tensorflow gpu和CUDA来加速训练。但它并没有加速。你知道吗

我首先安装了tensorfow gpu，如下所示： sudo pip3 install --upgrade tensorflow-gpu

然后，我安装CUDA un CUDNN如下： sudo pacman -S cuda cudnn

我最后检查了安装是否顺利： nvidia-smi

到目前为止，一切正常。但我随后启动了我的CNN的培训，我看不出在培训时间上有什么区别。你知道吗

CNN将MNIST分类如下：

• 1个conv层，最大池：28x28x1输入，14x14x6输出。你知道吗
• 1个最大池化转换层：14x14x6输入，7x7x16输出。你知道吗
• 1完全连接：7x7x16=784输入，128输出。你知道吗
• 1个完全连接：128英寸，10个输出。你知道吗

1epoch大约是21秒。 整个100个时代的训练时间大约是35分钟。你知道吗

我有：

• GPU->；GTX 850M
• CUDA->；版本10.2。你知道吗
• CPU->；i7 4700HQ
• 操作系统->；Manjaro

那么，你知道为什么我没有看到训练时间的改善吗？我需要在代码中添加什么吗？我以为tensorflow会自动选择CPU还是GPU，如果两者都可用？你知道吗