win10下安装Anaconda+Tensor Flow-GPU

前言

本文写于2019年12月(旧文搬运)。

本文主要介绍的是tensorflow-GPU版本的安装。GPU版本需要CUDA+cuDNN的支持,本文虽然没有介绍,但是是在已安装CUDA+cuDNN的基础上实行的。

CUDA+cuDNN相较起来坑是比较多的,尤其需要注意的是版本问题,一定要看对应版本不能盲目安装,否则会产生许多不必要的麻烦。

一、查看配置

  • 硬件:Nvidia的显卡(我的是GT940MX

    • 如何查看显卡型号?

      “我的电脑 ->(右键)管理 -> 设备管理器 -> 显示适配器”

      win10下安装Anaconda+Tensor Flow-GPU_第1张图片

  • tensorflow的版本选择:

    tensorflow 有两个版本:CPU 和 GPU。(我这里选择了GPU)

    我们先看看他们的全称:

    CPU (Central Processing Unit 中央处理器)

    GPU (Graphics Processing Unit 图形处理器)

    显然,CPU作为中央处理器,需要很强的通用性,它需要的是处理复杂的工作任务。而

    GPU的工作更为单一,计算量大,重复高。

    我们可以把CPU当作一位知识渊博的教授,而GPU就是一群小学生。但如果工作要求计算几千次的一百以内的加减乘除,显然,一群小学生比一位教授更为适合。

    所以GPU更适合矩阵运算,更适合神经网络的处理。

    !但是,GPU的选择是有硬件要求的,因为GPU版本需要CUDA和cuDNN的支持,要想安装GPU,首先要看一下自己的显卡是否支持CUDA。Nvidia的显卡可以在 https://developer.nvidia.com/cuda-gpus 查看自己的GPU是否支持CUDA。

    win10下安装Anaconda+Tensor Flow-GPU_第2张图片

基础要求满足了,就可以开始安装 tensorflow 了 。

二、安装Anaconda

Anaconda 是 python 的一个集成管理工具或系统。面对不同的项目环境需求,Anaconda 的便利就可以体现出来了,它可以创建多个不同的环境且互不干扰。

1. 下载与安装

下载地址:https://www.anaconda.com/download/

根据自己电脑的实际需求安装,我是windows X64。(备注:如果使用caffe需要注意,caffe 依赖于 boost 库,boost默认为 Python2,当然也可以用 python3 编译 boost库,我这里就不多介绍了。) win10下安装Anaconda+Tensor Flow-GPU_第3张图片

安装过程一路向下就好了。

win10下安装Anaconda+Tensor Flow-GPU_第4张图片

win10下安装Anaconda+Tensor Flow-GPU_第5张图片

win10下安装Anaconda+Tensor Flow-GPU_第6张图片

2. 配置环境变量

E:\Anaconda3
E:\Anaconda3\Scripts
E:\Anaconda3\Library\bin

将以上三个路径加入环境变量,需要根据自己的实际安装路径修改。

3. 查看安装结果

在cmd中如下输入“conda -V”,正确出现版本号就表示安装成功了。

4. 添加清华源

配置国内镜像可以使得速度快一些,这里添加的是清华源。

根据官网(https://mirror.tuna.tsinghua.edu.cn/help/anaconda/),修改用户目录下的.condarc文件:

channels:

  • defaults

show_channel_urls: true
default_channels:

custom_channels:
conda-forge: https://mirrors.tuna.tsinghua...
msys2: https://mirrors.tuna.tsinghua...
bioconda: https://mirrors.tuna.tsinghua...
menpo: https://mirrors.tuna.tsinghua...
pytorch: https://mirrors.tuna.tsinghua...
simpleitk: https://mirrors.tuna.tsinghua...

(Windows 用户无法直接创建名为.condarc的文件,可先执行conda config --set show_channel_urls yes生成该文件后再修改。

二、安装 tensorflow(GPU版本)

1. 创建conda环境

conda create -n [your_env_name] python=3.7

python版本可以根据实际需求更改,但是需要注意tensorflow支持的python版本:https://www.tensorflow.org/install/source_windows(以下为部分内容展示)

win10下安装Anaconda+Tensor Flow-GPU_第7张图片

(若是创建失败,提示“the following specifications were found to be incompatible with other”,可以先执行conda update conda

出现以下界面就表示创建环境成功了。

win10下安装Anaconda+Tensor Flow-GPU_第8张图片

2.激活环境

activate [your_env_name]

执行以上命令行进入刚刚创建的环境

3. 安装Anaconda基础包

因为我们是新建的“空包”,所以需要安装基础包。执行以下命令行:

conda install anconda

  • 注意:

    Anaconda 中各环境是相互隔离互不干扰的,所以在使用是需要注意当前使用的环境,如果在[your_env_tensorflow]环境中安装了tensorflow,使用的却是没有安装的[base]环境,自然是会报错的。这也是我们需要安装基础包的原因。

4. 安装tensorflow

  • CPU版本

    pip install --ignore-installed --upgrade tensorflow

  • GPU版本

    pip install --ignore-installed --upgrade tensorflow-gpu

这里着重安装的是GPU版本,CPU就不多做介绍了。当出现以下界面就表示安装成功了:

win10下安装Anaconda+Tensor Flow-GPU_第9张图片

其余 conda 的常用命令就不做赘述了。

5. 检验

win10下安装Anaconda+Tensor Flow-GPU_第10张图片

出现Adding visible gpu devices:0即表示GPU版本安装成功

参考:

https://cloud.tencent.com/developer/article/1165267?client=tim&ADUIN=1159622914&ADSESSION=1567594756&ADTAG=CLIENT.QQ.5603_.0&ADPUBNO=26933

你可能感兴趣的:(人工智能,tensorflow)