1,网上很多说是代码有问题,但是我是直接将cpu的代码复制过来,改了device,还是不行;
2,将 tensorflow-gpu,CUDA,cuDNN版本匹配起来 之后就解决了。
**解决方案:1,检查代码;2,安装正确的tensorflow-gpu,CUDA,cuDNN版本。
祝好!**
1,网上很多说是代码有问题,但是我是直接将cpu的代码复制过来,改了device,还是不行;
2,将 tensorflow-gpu,CUDA,cuDNN版本匹配起来 之后就解决了。
**解决方案:1,检查代码;2,安装正确的tensorflow-gpu,CUDA,cuDNN版本。
祝好!**