首页 > 其他 > 详细

安装TensorFlow踩的坑

时间:2019-03-30 21:27:43      阅读:273      评论:0      收藏:0      [点我收藏+]

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(注意版本)

安装TensorFlow踩的坑

原文:https://www.cnblogs.com/tccbj/p/10628844.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!