1,网上很多说是代码有问题,但是我是直接将cpu的代码复制过来,改了device,还是不行;
2,将 tensorflow-gpu,CUDA,cuDNN版本匹配起来 之后就解决了。
**解决方案:1,检查代码;2,安装正确的tensorflow-gpu,CUDA,cuDNN版本。
祝好!**
本文探讨了在使用GPU进行模型训练时遇到的不收敛问题,并提供了两种解决方案:一是检查代码确保没有错误;二是正确安装匹配版本的tensorflow-gpu、CUDA及cuDNN。
1,网上很多说是代码有问题,但是我是直接将cpu的代码复制过来,改了device,还是不行;
2,将 tensorflow-gpu,CUDA,cuDNN版本匹配起来 之后就解决了。
**解决方案:1,检查代码;2,安装正确的tensorflow-gpu,CUDA,cuDNN版本。
祝好!**
1981
1394

被折叠的 条评论
为什么被折叠?
