您的位置: 首页  >  文章  >  Win10和VS2017环境下安装tensorflow

Win10和VS2017环境下安装tensorflow

分类: 文章 • 2023-10-11 21:21:41
一、安装前准备
TensorFlow 有两个版本：CPU 版本和 GPU 版本。GPU 版本需要 CUDA 和 cuDNN 的支持，CPU 版本不需要。如果安装 GPU 版本，需要先确认显卡支持 CUDA（查看你的GPU是否支持CUDA）。我安装的是tensorflow-gpu 1.5.0rc1 版本。
tensorflow1.5.0版本需要安装CUDA 9.0 和cuDNN7(注意版本号，一定不要错了，安装前可以到GitHub看下要求)。传送门

二、安装步骤
  1. 下载python安装包，安装python，添加环境变量（Script目录也要加进去）
  2. 通过pip安装tensorflow（由于从国外网站下载相关软件包，速度较慢。可以指定国内源进行下载安装。此处通过 -i 选项指定清华源https://pypi.tuna.tsinghua.edu.cn/simple。 ）
    1. power shell输入“pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.5.0rc1 ”（通过==指定版本号）
    2. 等待安装完成，这个时候输入import tensorflow as tf会提示没有找到cuda的dll
  3. 下载安装cuda，下载地址。安装完成之后命令行输入”nvcc -V“会显示版本号。
  4. 下载cuDNN，下载地址，需要注册NVIDIA的账号。下载完成之后解压，添加环境变量即可
三、测试运行

$ python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print sess.run(a+b)
42
>>>
四、在vs2017中编写python
在vs2017安装程序中勾选python开发，并且将旁边的python本机开发勾选上，如下图所示。建议将anaconda勾选上，因为这个集成了很多科学计算所必需的库，能够避免很多依赖问题。
Win10和VS2017环境下安装tensorflow

走过的弯路：
之前安装了tensorflow1.4.0版本，安装了cuda9.1，无法正常使用。卸载tensorflow1.4.9时一个文件找不到，最后在目录中搜索到那个文件，发现他是在cache目录中的，删除或者重名cache目录即可。tensorflow1.5.0rc1对应的cuda是9.0版本，即使安装cuda9.1版本也不行。

参考资料：

相关推荐