安装TensorFlow踩的坑

2024年2月6日 70次阅读来源: 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（注意版本）

    原文作者：tensorflow
    原文地址: https://www.cnblogs.com/tccbj/p/10628844.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。