Window10上Tensorflow的安装(CPU和GPU版本)


Posted in Python onDecember 15, 2020

之前摸索tensorflow的时候安装踩坑的时间非常久,主要是没搞懂几个东西的关系,就在瞎调试,以及当时很多东西不懂,很多报错也一知半解的。这次重装系统后正好需要再配置一次,把再一次的经历记录一下。我的电脑是华为的matebook13,intel i5-8625U,MX250显卡,win10系统。(不得不吐槽很垃圾,只能满足测试测试调调代码的需求)

深度学习利用Tensorflow平台,其中的Keras Sequential API对新用户非常的友好,可以将各基础组件组合在一起来构建模型。

(官网: https://tensorflow.google.cn/?hl=zh-cn)

Window10上Tensorflow的安装(CPU和GPU版本)

Window10上Tensorflow的安装(CPU和GPU版本)

安装Tensorflow 分为 tensorflow_cpu 和 tensorflow_gpu版本

GPU就是用来渲染计算的,GPU版本计算性能是CPU的百倍之快。如果电脑没有独立显卡只能用CPU版本计算。

Window10上Tensorflow的安装(CPU和GPU版本)

CPU版本安装:

tensorflow_cpu版本只需要安装anaconda后在anaconda prompt里面pip install tensorflow_cpu==(版本号) 即可。安装anaconda的方法见GPU版本里面。

注意查下python和tensorflow_cpu适配的版本号。

Window10上Tensorflow的安装(CPU和GPU版本)

GPU版本安装

tensorflow_gpu版本安装大致分为三步:1、安装anaconda 2、安装cuda和cdunn 3、安装tensorflow_gpu。其中第一步和第二步的顺序可以调换,就是安装完了前面三个东西再安装tensorflow即可。

tensorflow最近出了2.0版本,和1.0大版本有一些区别。具体我还没有去了解,代码不一定兼容,需要注意一下。这三个步骤的版本需要格外的注意,一旦三个自己的版本互相不兼容或者和电脑的显卡不兼容,就用不了。所以安装前看看要安装的tensorflow版本。先查好显卡的算力,然后适配的相应版本再安装,否则踩坑要很久。

首先确定电脑所能支持的tensorflow版本,根据tensorflow官网所给配置,我们要去检查电脑gpu的cuda支持版本,再去对应下载python版本和tensorflow版本。

右键 桌面>NVIDIA控制面板>帮助>系统信息>组件

Window10上Tensorflow的安装(CPU和GPU版本)

上图说明我的显卡所支持的cuda版本为11.1(向下兼容)

Tensorflow配置window官网:https://tensorflow.google.cn/install/source_windows

Window10上Tensorflow的安装(CPU和GPU版本)

以我电脑为例,之前我用的是3.7.3的版本。这次重装系统后安装的是3.8.3。

cmd查看python版本:Window10上Tensorflow的安装(CPU和GPU版本)

我之前tensorflow用的是1.13的版本,这次因为已经安装了python3.8,因此打算安装2.0版本。如果仍需要低版本需要重新安装python较低的版本或者搭一个虚拟环境(但不是很建议)。

第一步,安装anaconda

anaconda会对应安装python环境,不一定要最新的, 最新的python版本不一定有兼容的cuda加速,有cuda加速也不一定支持电脑的显卡。

官网安装:https://www.anaconda.com/products/individual,之前版本如果官网没有可能需要找资源。

第二步,安装cuda工具包

(官网:https://developer.nvidia.com/zh-cn/cuda-toolkit)

CUDA是 NVIDIA 专为图形处理单元 (GPU) 上的通用计算开发的并行计算平台和编程模型。借助 CUDA,开发者能够利用 GPU 的强大性能显著加速计算应用。在经 GPU 加速的应用中,工作负载的串行部分在 CPU 上运行,且 CPU 已针对单线程性能进行优化,而应用的计算密集型部分则以并行方式在数千个 GPU 核心上运行。使用 CUDA 时,开发者使用主流语言(如 C、C++、Fortran、Python 和 MATLAB)进行编程,并通过扩展程序以几个基本关键字的形式来表示并行性。由于tensorflow最高版本对应的是cuda10.1版本,那下载cuda10.1即可。安装包链接:https://developer.nvidia.com/cuda-toolkit-archive

Window10上Tensorflow的安装(CPU和GPU版本)

Window10上Tensorflow的安装(CPU和GPU版本)

Window10上Tensorflow的安装(CPU和GPU版本)

CUDA的下载需要挂载VPN,否则下下来只有1kb (好像后面又不一定需要VPN,可以直接下载试试)

自定义安装,安装的东西全选了,尽量不要改安装位置

Window10上Tensorflow的安装(CPU和GPU版本)

安装完之后安装cuDNN, cuDNN是用于深度神经网络的GPU加速库.注意安装的版本,选择自身cuda版本对应的cudnn下载。

官网:https://developer.nvidia.com/rdp/cudnn-archive ,cuDNN的下载需要注册官网的账号

Window10上Tensorflow的安装(CPU和GPU版本)

下载好的cudnn文件解压后,将文件夹内的文件放到cuda对应文件夹下,注意:是文件夹内的文件,而不要直接复制替换文件夹

Window10上Tensorflow的安装(CPU和GPU版本)

第三步,安装tensorflow—gpu

在anaconda prompt里面安装:pip install tensorflow-gpu , 后面可以指定版本号,下载慢挂载一个镜像源 -i https://pypi.tuna.tsinghua.edu.cn/simple

Window10上Tensorflow的安装(CPU和GPU版本)

安装完成:

Window10上Tensorflow的安装(CPU和GPU版本)

测试是否安装成功:

import tensorflow as tf
tf.test.is_gpu_available()

Window10上Tensorflow的安装(CPU和GPU版本)

Window10上Tensorflow的安装(CPU和GPU版本)

True即表示安装成功了。

查看版本号:Window10上Tensorflow的安装(CPU和GPU版本)

至此安装成功,可以跑一个小程序测试一下。

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
 
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:', tf.test.is_gpu_available())

显示“GPU True”, 也即代表GPU版本安装成功。

经过几天调了一下代码之后发现tensorflow2相对与1还是有挺多改动的,1里面能够运行的代码可能2里面需要一定的修改。

关于Tensorflow2和1上面keras的一些区别可以搜一下相关的资料。

到此这篇关于Window10上Tensorflow的安装(CPU和GPU版本)的文章就介绍到这了,更多相关Window10安装Tensorflow 内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
基于python爬虫数据处理(详解)
Jun 10 Python
R语言 vs Python对比:数据分析哪家强?
Nov 17 Python
详解Python 装饰器执行顺序迷思
Aug 08 Python
Python实现图片添加文字
Nov 26 Python
Python线程障碍对象Barrier原理详解
Dec 02 Python
numpy.transpose()实现数组的转置例子
Dec 02 Python
Python如何使用argparse模块处理命令行参数
Dec 11 Python
为什么黑客都用python(123个黑客必备的Python工具)
Jan 31 Python
python中的selenium安装的步骤(浏览器自动化测试框架)
Mar 17 Python
给Django Admin添加验证码和多次登录尝试限制的实现
Jul 26 Python
Pycharm2020.1安装中文语言插件的详细教程(不需要汉化)
Aug 07 Python
一篇文章弄懂Python关键字、标识符和变量
Jul 15 Python
详解python中的异常捕获
Dec 15 #Python
python 6种方法实现单例模式
Dec 15 #Python
Ubuntu16安装Python3.9的实现步骤
Dec 15 #Python
Python爬虫开发与项目实战
Dec 16 #Python
python中reload重载实例用法
Dec 15 #Python
python 实现socket服务端并发的四种方式
Dec 14 #Python
linux centos 7.x 安装 python3.x 替换 python2.x的过程解析
Dec 14 #Python
You might like
PHP 实现代码复用的一个方法 traits新特性
2015/02/22 PHP
PHP设计模式之观察者模式定义与用法示例
2018/08/04 PHP
innerHTML,outerHTML,innerTEXT三者之间的区别
2007/01/28 Javascript
json 实例详细说明教程
2009/10/31 Javascript
javascript是怎么继承的介绍
2012/01/05 Javascript
Javascript处理DOM元素事件实现代码
2012/05/23 Javascript
JQquery的一些使用心得分享
2012/08/01 Javascript
flash调用js中的方法,让js传递变量给flash的办法及思路
2013/08/07 Javascript
jQuery功能函数详解
2015/02/01 Javascript
如何使用jQuery技术开发ios风格的页面导航菜单
2015/07/29 Javascript
Js调用Java方法并互相传参的简单实例
2016/08/11 Javascript
JS实现图片延迟加载并淡入淡出效果的简单方法
2016/08/25 Javascript
详解JS中的立即执行函数
2017/02/24 Javascript
jQuery与vue实现拖动验证码功能
2018/01/30 jQuery
使用vue的transition完成滑动过渡的示例代码
2018/06/25 Javascript
详解vue中this.$emit()的返回值是什么
2019/04/07 Javascript
Vue自定义全局弹窗组件操作
2020/08/11 Javascript
在antd Form表单中select设置初始值操作
2020/11/02 Javascript
python3编写C/S网络程序实例教程
2014/08/25 Python
Python3使用xml.dom.minidom和xml.etree模块儿解析xml文件封装函数的方法
2019/09/23 Python
Python高级property属性用法实例分析
2019/11/19 Python
Python+OpenCV 实现图片无损旋转90°且无黑边
2019/12/12 Python
使用python自动追踪你的快递(物流推送邮箱)
2020/03/17 Python
Python Tricks 使用 pywinrm 远程控制 Windows 主机的方法
2020/07/21 Python
python tkinter实现连连看游戏
2020/11/16 Python
html5组织文档结构_动力节点Java学院整理
2017/07/11 HTML / CSS
html5中svg canvas和图片之间相互转化思路代码
2014/01/24 HTML / CSS
基于HTML5代码实现折叠菜单附源码下载
2015/11/27 HTML / CSS
日本最大化妆品和美容产品的综合口碑网站:cosme shopping
2019/08/28 全球购物
Linux的主要特性
2016/09/03 面试题
2014年预备党员学习两会心得体会
2014/03/17 职场文书
民主生活会对照检查材料(统计局)
2014/09/21 职场文书
2014年流动人口工作总结
2014/11/26 职场文书
法人代表证明书范本
2015/06/18 职场文书
Python中相见恨晚的技巧
2021/04/13 Python
Python如何加载模型并查看网络
2022/07/15 Python