1、TensorFlow的版本要和对应的cuda,cudnn版本都对应上,装了cuda 9.0和cudnn 7.5,TensorFlow默认装了最新版1.13.1,一直出现 DLL error: 找不到指定模块。装了各种VS2015、2017都没用,最后发现是TensorFlow版本太高,需要cuda10.0支持,所以要卸了装1.12版本。
2、cuda版本应该和显卡版本关系不大,显卡显示的是cuda9的版本,应该也可以装cuda 10.0 和 TensorFlow 1.13,不过没有去尝试
3、装cuda和cudnn只要两个版本对应好,别的都按照网上方法,在官网上下对应版本,再复制cudnn到cuda对应目录下,加入环境变量中就OK了
4、在conda里新建一个 tensorflow-gpu 环境,命令:conda create -n tensorflow-gpu python=3.5
以上几步可以同时进行
5、activate tensorflow-gpu 进入环境,先升级pip,命令:python -m pip install --upgrade pip
6、用 pip 直接安装TensorFlow,命令 pip install tensorflow-gpu==1.12(注意版本)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。