win10 TensorFlow GPU环境搭建记录

win10 TensorFlow GPU环境搭建记录

By PorYoung
2018-10-17
在CSDN查看

机器配置

  • win 10(64bit)
  • GTX 960m

本次搭建所需文件

  • VS2017
  • CUDA
  • cuDNN
  • Anaconda

配置过程

  1. 安装(已安装则选择修改单个组件)并配置vs2017,添加如下组件:
    • VC++ 2017版本 15.4 v14.11工具集
    • 用于CMake的Visual C++工具
    • 适用于桌面的VC++ 2015.3 V14.00(V140)工具集
      • 首次安装CUDA时出现Visual Studio Integration无法安装,添加该组建后重新安装成功,不知道是不是受它的影响
  2. 安装Anaconda
    1. Anaconda包含了conda、Python等180多个科学包及其依赖项,功能十分强大
    2. 下载地址:https://www.anaconda.com/download/
    3. 此处选择了Python 3.7版本
      1. 后来发现目前Tensorflow不支持pythob 3.7版本,在安装Tensorflow时会提示找不到包
      2. 此外,安装部分版本不兼容的CUDA、cuDNN也会导致安装失败
      3. 对应版本解决方案:https://github.com/fo40225/tensorflow-windows-wheel,查看大神编译的各种版本tensorflow的地址,此次配置的为蓝色框线内的版本,Anaconda可以在后续修改环境为python 3.6
    4. 安装
      • 第一项可选可不选,不选的话需要自行配置环境变量
      • 第一项未选需要配置的环境变量
  3. 安装CUDA(注意版本:exclamation:)
    1. 安装前注意
      1. GTX 960m运算能力为5.0,本次选择最新v10.0
      2. 网上许多资料都表示存在tensorflow、CUDA和cuDNN版本不兼容无法支持等问题,未一一测试,均选择当时最新版本。比较谨慎的,可以参考相关博文。
    2. 查看显卡支持:https://developer.nvidia.com/cuda-gpus
    3. 选择对应版本下载:https://developer.nvidia.com/cuda-toolkit-archive
    4. 安装
      1. 选择自定义安装
      2. 一般不需要安装GeForce Experience
    5. 安装可能存在的问题
      1. Visual Studio Intergration无法安装
        > 可能的解决方法
        > 1. 参考安装VS2017的过程,可能缺少组件
        > 2. 参考教程:CUDA安装失败解决方法
  4. 下载cuDNN(注意版本:exclamation:),需要注册
    1. 下载链接:https://developer.nvidia.com/rdp/cudnn-download
    2. 此处选择和CUDA v10.0搭配的cudnn-10.0-windows10-x64-v7.3.1.20
    3. 解压到CUDA安装根目录,共三个文件夹:binincludelib
  5. 安装Tensorflow运行环境
    1. 打开Anaconda prompt
    2. 配置清华仓库镜,输入指令:
      conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
      conda config --set show_channel_urls yes
    3. 创建运行环境,名称为tensorflow-gpu,python版本为3.6
      conda create -n tensorflow-gpu python=3.6
      名称作为与其他环境隔离的标志,版本可自行更换
    4. 激活并进入该环境
      activate tensorflow-gpu
      其他指令可以参考Anaconda教程
    5. 升级pip
      python -m pip install --upgrade pip
    6. 安装相关依赖包
      • 如果确定所安装的版本兼容,可以直接安装
        pip install --ignore-installed --upgrade tensorflow-gpu
      • 如果使用的是编译的tensorflow版本则需要进入下载目录进行安装,如进入D:\Files目录,有从github下载的编译版本tensorflow_gpu-1.11.0-cp36-cp36m-win_amd64.whl,执行安装命令即可
        pip install tensorflow_gpu-1.11.0-cp36-cp36m-win_amd64.whl
    7. 测试Tensorflow
      1. 进入环境,运行python
      2. 键入
        import tensorflow as tf
      3. 未报错则安装成功
      4. 可能存在的问题
        1. 报DLL找不到模块,可能是版本选择的问题
        2. 其他问题尚未可知

参考文章

  1. Win10下Tensorflow(GPU版)安装趟坑实录
  2. Win10 64 位Tensorflow-gpu安装(VS2017+CUDA9.2+cuDNN7.1.4+python3.6.5)
  3. Win10+VS2017+CUDA9.2 安装调试笔记

本文链接:win10 TensorFlow GPU环境搭建记录

转载声明:本站文章若无特别说明,皆为原创,转载请注明来源:PorYoung,谢谢!^^

暂无评论

发送评论 编辑评论


				
|´・ω・)ノ
ヾ(≧∇≦*)ゝ
(☆ω☆)
(╯‵□′)╯︵┴─┴
 ̄﹃ ̄
(/ω\)
∠( ᐛ 」∠)_
(๑•̀ㅁ•́ฅ)
→_→
୧(๑•̀⌄•́๑)૭
٩(ˊᗜˋ*)و
(ノ°ο°)ノ
(´இ皿இ`)
⌇●﹏●⌇
(ฅ´ω`ฅ)
(╯°A°)╯︵○○○
φ( ̄∇ ̄o)
ヾ(´・ ・`。)ノ"
( ง ᵒ̌皿ᵒ̌)ง⁼³₌₃
(ó﹏ò。)
Σ(っ °Д °;)っ
( ,,´・ω・)ノ"(´っω・`。)
╮(╯▽╰)╭
o(*////▽////*)q
>﹏<
( ๑´•ω•) "(ㆆᴗㆆ)
😂
😀
😅
😊
🙂
🙃
😌
😍
😘
😜
😝
😏
😒
🙄
😳
😡
😔
😫
😱
😭
💩
👻
🙌
🖕
👍
👫
👬
👭
🌚
🌝
🙈
💊
😶
🙏
🍦
🍉
😣
Source: github.com/k4yt3x/flowerhd
颜文字
Emoji
小恐龙
花!
上一篇
下一篇