关于环境搭建网上有很多教程,在此贴一下我配时参考的教程:https://blog.csdn.net/qq_30611601/article/details/79067982
在这里主要写一下我躺的雷:
- tensorflow-gpu+cuda+cudnn 版本之间的联系,最开始没注意这个问题,导致后面出了很多莫名其妙的错误。以后配置参照这张表即可:
如欲了解更多版本匹配,可参考链接:https://blog.csdn.net/omodao1/article/details/83241074
2. 关于tensorflow-gpu版本过高,如何降版本。
先pip uninstall tenserflow-gpu(卸载)
卸载成功以后,输入:pip install --user tensorflow-gpu==1.3.0(版本号根据实际情况选择)-i https://pypi.doubanio.com/simple/
即可。
可参考链接:http://baijiahao.baidu.com/s?id=1604501192403223852&wfr=spider&for=pc