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的@property装饰器的用法
Apr 28 Python
使用Python脚本将Bing的每日图片作为桌面的教程
May 04 Python
21行Python代码实现拼写检查器
Jan 25 Python
python批量替换多文件字符串问题详解
Apr 22 Python
Python3多进程 multiprocessing 模块实例详解
Jun 11 Python
解决Python下imread,imwrite不支持中文的问题
Dec 05 Python
Opencv+Python实现图像运动模糊和高斯模糊的示例
Apr 11 Python
twilio python自动拨打电话,播放自定义mp3音频的方法
Aug 08 Python
python调用支付宝支付接口流程
Aug 15 Python
python实现单张图像拼接与批量图片拼接
Mar 23 Python
Python基础知识之变量的详解
Apr 14 Python
总结Pyinstaller打包的高级用法
Jun 28 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
跟我学小偷程序之成功偷取首页(第三天)
2006/10/09 PHP
两种php调用Java对象的方法
2006/10/09 PHP
yii框架表单模型使用及以数组形式提交表单数据示例
2014/04/30 PHP
php实现扫描二维码根据浏览器类型访问不同下载地址
2014/10/15 PHP
PHP实现对png图像进行缩放的方法(支持透明背景)
2015/07/15 PHP
php版微信自动获取收货地址api用法示例
2016/09/22 PHP
动态改变textbox的宽高的js
2006/10/26 Javascript
javascript转换静态图片,增加粒子动画效果
2015/05/28 Javascript
JavaScript获取function所有参数名的方法
2015/10/30 Javascript
js操作数组函数实例小结
2015/12/10 Javascript
javascript瀑布流布局实现方法详解
2016/02/17 Javascript
JavaScript数据结构之二叉查找树的定义与表示方法
2017/04/12 Javascript
详解vue中点击空白处隐藏div的实现(用指令实现)
2018/04/19 Javascript
layui 设置table 行的高度方法
2018/08/17 Javascript
微信小程序实现保存图片到相册功能
2018/11/30 Javascript
Vue过渡效果之CSS过渡详解(结合transition,animation,animate.css)
2020/02/05 Javascript
Vue插件之滑动验证码用法详解
2020/04/05 Javascript
vue实现虚拟列表功能的代码
2020/07/28 Javascript
JavaScript实现移动小精灵的案例代码
2020/12/12 Javascript
[10:21]DOTA2-DPC中国联赛 正赛 PSG.LGD vs Aster 选手采访
2021/03/11 DOTA
python对于requests的封装方法详解
2019/01/03 Python
django的模型类管理器——数据库操作的封装详解
2020/04/01 Python
keras中模型训练class_weight,sample_weight区别说明
2020/05/23 Python
如何使用Python处理HDF格式数据及可视化问题
2020/06/24 Python
关于tensorflow softmax函数用法解析
2020/06/30 Python
用友笔试题目
2016/10/25 面试题
文秘专业应届生求职信范文
2013/11/14 职场文书
毕业生护理专业个人求职信范文
2014/01/04 职场文书
竞选大队长演讲稿
2014/04/29 职场文书
学习保证书范文
2014/04/30 职场文书
机电一体化专业求职信
2014/07/22 职场文书
2014年医药代表工作总结
2014/11/22 职场文书
元旦晚会主持词开场白
2015/05/28 职场文书
2016春节放假通知范文
2015/08/18 职场文书
2016庆祝教师节新闻稿
2015/11/25 职场文书
适合毕业生创业的项目怎么找?
2019/08/08 职场文书