安装TensorFlow踩的坑

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

猜你喜欢

转载自www.cnblogs.com/tccbj/p/10628844.html