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写入CSV文件的方法
Jul 08 Python
Python实现按逗号分隔列表的方法
Oct 23 Python
python生成器与迭代器详解
Jan 01 Python
python3实现zabbix告警推送钉钉的示例
Feb 20 Python
Django处理多用户类型的方法介绍
May 18 Python
python多进程并行代码实例
Sep 30 Python
详解有关PyCharm安装库失败的问题的解决方法
Feb 02 Python
对Tensorflow中Device实例的生成和管理详解
Feb 04 Python
keras 实现轻量级网络ShuffleNet教程
Jun 19 Python
keras 简单 lstm实例(基于one-hot编码)
Jul 02 Python
实例讲解Python 迭代器与生成器
Jul 08 Python
python实现猜拳游戏项目
Nov 30 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
mysql中存储过程、函数的一些问题
2007/02/14 PHP
对squid中refresh_pattern的一些理解和建议
2009/04/17 PHP
php缓存技术详细总结
2013/08/07 PHP
php简单smarty入门程序实例
2015/06/11 PHP
项目中应用Redis+Php的场景
2016/05/22 PHP
PHP命名空间namespace用法实例分析
2016/09/27 PHP
JavaScript 参数中的数组展开 [译]
2012/09/21 Javascript
js获取本机的外网/广域网ip地址完整源码
2013/08/12 Javascript
node.js中的fs.rmdir方法使用说明
2014/12/16 Javascript
JavaScript中数组继承的简单示例
2015/07/29 Javascript
js+CSS实现模拟华丽的select控件下拉菜单效果
2015/09/01 Javascript
JavaScript实战之带收放动画效果的导航菜单
2016/08/16 Javascript
js数组操作方法总结(必看篇)
2016/11/22 Javascript
JS使用插件cryptojs进行加密解密数据实例
2017/05/11 Javascript
在Vue环境下利用worker运行interval计时器的步骤
2019/08/01 Javascript
JQuery获得内容和属性方法解析
2020/05/30 jQuery
[01:15:29]DOTA2上海特级锦标赛主赛事日 - 3 胜者组第二轮#2Secret VS EG第三局
2016/03/04 DOTA
Python简单实现阿拉伯数字和罗马数字的互相转换功能示例
2018/04/17 Python
Python 3 实现定义跨模块的全局变量和使用教程
2019/07/07 Python
tensorflow ckpt模型和pb模型获取节点名称,及ckpt转pb模型实例
2020/01/21 Python
python3.6使用SMTP协议发送邮件
2020/05/20 Python
手把手教你从PyCharm安装到激活(最新激活码),亲测有效可激活至2089年
2020/11/25 Python
Python爬虫之Selenium实现关闭浏览器
2020/12/04 Python
详解移动端h5页面根据屏幕适配的四种方案
2020/04/15 HTML / CSS
美国专注于健康商品的网站:eVitamins
2017/01/23 全球购物
澳大利亚百货公司:David Jones
2018/02/08 全球购物
信息管理专业推荐信
2013/10/29 职场文书
房屋出租协议书
2014/04/10 职场文书
施工工地安全标语
2014/06/07 职场文书
联片教研活动总结
2014/07/01 职场文书
向雷锋同志学习倡议书
2015/04/27 职场文书
工程款申请报告
2015/05/15 职场文书
导游词之蜀山胜景瓦屋山
2019/11/29 职场文书
DjangoRestFramework 使用 simpleJWT 登陆认证完整记录
2021/06/22 Python
angular4实现带搜索的下拉框
2022/03/25 Javascript