运行tensorflow基本程序mnist.py

来源:互联网 发布:什么照相软件好 编辑:程序博客网 时间:2024/05/29 05:55

进入root后,运行,出现
AttributeError:’Cycler’ object has no attribute ‘change_key’
经查询,cycler版本过低,利用
sudo -H python3 -m pip install –upgrade cycler
解决问题,重复运行,出现
2017-06-29 15:15:02.988781: E tensorflow/stream_executor/cuda/cuda_dnn.cc:352] Loaded runtime CuDNN library: 5005 (compatibility version 5000) but source was compiled with 5110 (compatibility version 5100). If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.
2017-06-29 15:15:02.988853: F tensorflow/core/kernels/conv_ops.cc:671] Check failed: stream->parent()->GetConvolveAlgorithms(&algorithms)

发现是cudnn版本不对,经过查询将cudnn5.0升级至5.1,

运行成功

原创粉丝点击