在Windows 10平台上安装TensorFlow

最新推荐文章于 2021-12-15 16:45:21 发布

Doke_Design

最新推荐文章于 2021-12-15 16:45:21 发布

阅读量325

点赞数

文章标签： TensorFlow

本文链接：https://blog.csdn.net/pinton_tech/article/details/80218753

版权

TensorFlow的版本

仅支持 CPU 的 TensorFlow。如果您的系统没有 NVIDIA® GPU，就必须安装此版本。请注意，此版本的 TensorFlow 通常更容易安装（用时通常在 5 或 10 分钟内），所以即使您拥有 NVIDIA GPU，我们也建议先安装此版本。预先构建的二进制文件将使用 AVX 指令。
支持 GPU 的 TensorFlow。TensorFlow 程序在 GPU 上的运行速度通常要比在 CPU 上快得多。因此，如果您的系统配有满足以下所示先决条件的 NVIDIA® GPU，并且您需要运行性能至关重要的应用，则最终应安装此版本。

如果你的PC不是NVIDIA显卡，请安装CPU版本的，如果是英伟达的显卡，请确定它的CUDA运算能力大于等于3.0。参考查阅CUBA计算能力查询，如果性能不够，请使用CPU版本，如果达标，则使用GPU版本。

CUDA：解压后运行安装程序，安装完成后系统变量会自动添加，验证是否成功的方法：右击屏幕“”，选择“Windows PowerShell（管理员）”，输入 nvcc -V，如正常出现版本号，安装成功。
cudnn：解压到一个位置，再把 (对应的位置)\cudnn-9.0-windows10-x64-v7.1\cuda\bin 复制，打开PowerShall，输入 sysdm.cpl ，ENTER打开了”系统属性“,点击”高级“->”环境变量“，选中下边系统变量的"Path”，“编辑”，
“新建”，并把地址粘贴上去，依次点击确定。
Anaconda：运行安装程序，注意过程中选中第一个框（最新版本会红色提示，仍然选中）

打开PowerShell或者在开始菜单找到

输入以下命令：

检测Anaconda是否安装正常，如出现版本号则正常。

conda --version

搭建Python 3.5

conda create --name tensorflow python=3.5

激活

activate tensorflow

安装TensorFlow（二选一）:

GPU版本：

pip3 install --upgrade tensorflow-gpu

CPU版本：

pip3 install --upgrade tensorflow

然后会有一些进度条，如果期间网速过慢，可能会出现红色的报错，这时重新输入安装的命令，重新开始即可。

为了模拟以后打开的情况，现在将PowerShell关闭。

打开Anaconda自带的Anaconda Prompt，不要使用Windows PowerShell。

输入

activate tensorflow

Python

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

如图所示，则安装成功。

新建项目后，File->Settings->Project;Python Project->Project Interpreter

点击右侧面板中的下拉菜单，选择Show All...

在弹出的面板中，点右侧的“+”，弹出另一个面板，按下图

选择安装路径，我的是C:\Users\账户名\AppData\Local\conda\conda\envs\tensorflow\python.exe，供大家参考。

选中Make available to all projects，OK即可。

重新回到这个面板，选择刚刚添加的TensorFlow，OK。

同样的测试语句

import tensorflow as tf

hello = tf.constant("Hello!TensorFlow")
sess = tf.Session()
print(sess.run(hello))

全部复制到PyCharm的代码工作区，点击右上角运行，

成功！

关注