当前位置:   article > 正文

如何利用GPU训练keras模型_keras使用gpu训练

keras使用gpu训练

1.如何利用GPU训练keras模型

(我当时是在服务器上的Linux系统下只有一块gpu)

改之前:

conda install tensorflow-gpu==1.15.0

pip install keras==2.2.4

然后用nvidia-smi看训练的过程有没有使用gpu

显示 No running processse found

修改:

conda install tensorflow-gpu==1.15.0

conda install keras-gpu==2.2.4

然后用nvidia-smi看训练的过程有没有使用gpu

修改之前我也看了很多csdn里的文章,最终是采用这位博主的经验:

解决keras不能用GPU加速的问题。_脆皮茄条的博客-CSDN博客_keras无法调用gpu

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号