Win10环境下安装TensorFlow

TensorFlow是机器学习中使用很多的一个框架,最近开始学习TensorFlow,记录一下安装的过程。

一、TensorFlow的安装

Python的安装此处不再赘述。在装好Python和pip的前提下(笔者Python的版本为3.7),首先“win+R”输入“cmd”打开命令提示符。输入pip install tensorflow,此时pip开始下载TensorFlow,这一步时间可能较长,请耐心等待。此处可能遇到需更新pip的情况,将弹出的更新命令进行复制粘贴后运行更新pip,再次执行pip install tensorflow命令直到成功。这便是TensorFlow CPU版本的安装。
TensorFlow GPU版的安装与上述类似。“win+R”输入“cmd”打开命令提示符。输入pip install tensorflow-gpu,耐心等待安装成功。

二、Cuda和cuDNN的安装

TensorFlow GPU版的安装还需要安装Cuda和cuDNN。此处Cuda和cuDNN的版本极其重要,绝大多数的安装错误都是因为安装的版本不对。还是打开“win+R”输入“cmd”打开命令提示符之后输入python,进入Python环境后输入import tensorflow,可以看到报错,原因是缺少cuda,如下图所示。
在这里插入图片描述
可以看到,系统提示安装cuda的版本为10.0.到NVIDIA官网https://developer.nvidia.com/cuda-toolkit-archive找到对应版本进行下载。运行下载下来的exe文件,如果嫌麻烦的话一路按默认选项点下一步就可以了。或者选择自定义。
在这里插入图片描述
取消勾选GeForce Experience,点下一步。
在这里插入图片描述
选择安装路径,建议默认(没有尝试安装到其他路径)。点击下一步进行安装。这一步的安装时间也比较长,耐心等待。
在这里插入图片描述
安装好cuda后,再次重复上述“import tensorflow”的操作。发现系统提示应该安装cuDNN,并给出了相应的版本。如图所示即64位7.0版本,到NVIDIA官网https://developer.nvidia.com/rdp/cudnn-archive找到对应版本进行下载。
在这里插入图片描述
将下载下来的压缩文件进行解压,发现cuda文件夹下有3个文件夹。
在这里插入图片描述
将这三个文件夹拷贝到cuda的安装目录下,比如我按默认路径安装的是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0。重名文件进行替换。
在这里插入图片描述
再次重复进行上述“import tensorflow”操作,发现程序没有报错。
在这里插入图片描述

三、TensorFlow安装的测试

“win+R”输入“cmd”后输入“python”进入python环境后输入如下代码(或在pycharm等测试也可以)

import tensorflow as tf
hello = tf.constant('hello,tensorf')
sess = tf.Session()
print(sess.run(hello))

进行测试,如果输出b’hello,tensorf’,则安装成功。
下图为命令提示符界面的结果。
在这里插入图片描述
pycharm的结果如下图所示。
在这里插入图片描述
参考文献
【1】风吹过01.win10环境下安装tensorflow.https://blog.csdn.net/qq_24800377/article/details/82856360
【2】W,Haixin.Win10下Tensorflow(GPU版)安装趟坑实录.https://blog.csdn.net/weixin_39290638/article/details/80045236
【3】_Yucen.TensorFlow测试程序.https://blog.csdn.net/qq_14997473/article/details/80511135

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值