Windows10配置TensorFlow-GPU深度学习环境

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/u010620587/article/details/89335182

一、 我的电脑配置
Intel® Xeon® W-2123 CPU @3.60GHz
内存32GB操作系统Windows10 x64
显卡 NVIDIA Quadro P2000
磁盘驱动 Samsung SSD 860 EVO 1TB
二、 Windows10下安装TensorFlow-gpu
(一) 安装Anaconda
https://www.anaconda.com/distribution/#download-section
下载Anaconda 2019.03 for Windows Installer,自带python3.7,安装过程中下列对话框第一项要打勾
在这里插入图片描述
(二) 安装VS2017(自带PYTHON3.6)
这个可能不是必要的,为了顺利安装CUDA做准备
这里面可选择性很强,我安装了跟基本的模块和与python有关的模块
后面参考了CSDN多篇教程(现在都找不到了……)
有人说TensorFlow1.13不支持python3.7,为了避免这个问题,我直接又装了python3.6
(三) 安装CUDA

  1. 查询GPU支持的CUDA版本
    右键桌面选择NVIDIA控制面板,左下角选择系统信息,在组件里面查看支持的是CUDA10.0
    在这里插入图片描述
  2. 下载对应版本的CUDA安装包
    https://developer.nvidia.com/cuda-toolkit-archive
    在这里插入图片描述
    按照步骤一步一步安装,自定义安装左右组件(下图我安装时只有三个选项)
    在这里插入图片描述
  3. 安装cudnn补丁
    下载对应的cudnn10.0,压缩包解压后,有三个文件夹,把三个文件夹拷贝到cuda的安装目录下,如果cuda安装的时候用的是默认路径,则为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
    在这里插入图片描述
  4. 测试CUDA
    打开cmd命令行,输入nvcc -V,看到下图内容:
    在这里插入图片描述
    我测试遇到问题,找不到NVCC的定义,最后添加两个环境变量解决的该问题:右键此电脑,进入系统属性,点击环境变量,在系统变量里面新建变量CUDA_bin和CUDA_libnvvp,值分别是安装路径下的对应位置,自己查一下就知道了。
    在这里插入图片描述
  5. 安装TensorFlow
    用pip安装,打开Anaconda Prompt,依次输入以下内容:
    输入:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
    输入:conda config --set show_channel_urls yes
    注释:配置清华仓库镜像
    输入:conda create -n tensorflow-gpu python=3.6
    注释:配置运行环境
    自动显示需下载和安装的内容## Package Plan ##
    输入:y
    自动开始下载和安装
    显示: ……
    Preparing transaction: done
    Verifying transaction: done
    Executing transaction: done
    输入:conda activate tensorflow-gpu
    注释:激活进入环境
    输入:python -m pip install --upgrade pip
    注释:升级pip
    显示:
    Cache entry deserialization failed, entry ignored
    Collecting pip
    Cache entry deserialization failed, entry ignored
    Downloading https://files.pythonhosted.org/packages/d8/f3/413bab4ff08e1fc4828dfc59996d721917df8e8583ea85385d51125dceff/pip-19.0.3-py2.py3-none-any.whl (1.4MB)
    100% |████████████████████████████████| 1.4MB 91kB/s
    Installing collected packages: pip
    Found existing installation: pip 9.0.1
    Uninstalling pip-9.0.1:
    Successfully uninstalled pip-9.0.1
    Successfully installed pip-19.0.3

输入:pip install --ignore-installed --upgrade tensorflow-gpu
在这里插入图片描述
(tensorflow-gpu) C:\Users\user>
输入:python
输入:
import tensorflow as tf
hello = tf.constant(‘Hello, TensorFlow!’)
sess = tf.Session()
输入:print(sess.run(hello))
显示:b’Hello, TensorFlow!’
成功。
在这里插入图片描述
参考:
https://blog.csdn.net/liuyong5573/article/details/85472808

展开阅读全文

没有更多推荐了,返回首页