深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南

本人小白,应课题学习要求,在经历了一周多的电脑组装后,终于可以亲手搭建自己的深度学习环境了。本文参考了大量的其他博文,如有不明白的地方,多多跳转参考链接。其实如果你会使用Anaconda的话,很多安装包都不用提前下载,直接pip在线安装,包括你在使用Python写代码的过程中如果缺少模块,都可以pip在线安装

  • 系统配置

  1. 操作系统(OS):Win10 x64
  2. 处理器(CPU):Intel(R) Core(TM) i7-2600 CPU @3.40GHz 四核八线程
  3. 内存(RAM):22.0 GB
  4. 显卡(GPU):NVIDIA GeForce GTX 1060 6GB
  5. 驱动程序版本:221.14.1881(418.81)
     
  • 安装包关系

参考链接:https://blog.51cto.com/12240152/2053144

深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第1张图片
 

  • 环境适配

  1. Anaconda(Python环境) 参考链接:https://tensorflow.google.cn/install/pip

    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第2张图片

    本文选用Anaconda 3-5.1.0-Windows-x86_64 默认安装Python3.6

  2. CUDA Toolkit  参考链接:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第3张图片

    本文选用CUDA 10.0.130

  3. cuDNN 参考链接:https://developer.nvidia.com/rdp/cudnn-download

    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第4张图片

    本文选用cuDNN 7.5.0 for CUDA 10.0

  4. TensorFlow  参考链接:https://blog.csdn.net/sinat_23619409/article/details/84202596

    本文选用TensorFlow 1.13.1 cp36

  • 安装步骤

安装参考链接:

https://www.jianshu.com/p/034a628ed45e

https://blog.csdn.net/weixin_38285131/article/details/79416592

  1. 安装Anaconda 3-5.3.1-Windows-x86_64

    下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

    -安装Anaconda

    -添加Anaconda 环境变量 参考链接:https://blog.csdn.net/z2858539804/article/details/81407456

    -创建TensorFlow环境 参考链接:https://blog.csdn.net/Kyle_2017/article/details/78257051?locationNum=2&fps=1

    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第5张图片

    -测试 Python

    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第6张图片

  2. 安装CUDA 10.0.130

    下载地址:https://developer.nvidia.com/cuda-toolkit-archive

    -安装VS2015(非必要)

    -安装CUDA

    -检查CUDA 参考链接:https://blog.csdn.net/qq_33856151/article/details/79295086

    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第7张图片

  3. 添加 cuDNN

    下载地址:https://developer.nvidia.com/rdp/cudnn-download

    -cuDNN解压后,把文件复制到CUDA对应目录

  4. 安装tensorflow

    参考链接:https://blog.csdn.net/weixin_38285131/article/details/79416592

    -创建python环境

    打开Anaconda Prompt输入:conda create -name tensorflow-gpu python=3.6.4 

    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第8张图片

    -激活tensorflow安装环境

    输入activate tensorflow

    -在线安装tensorflow

    输入pip install tensorflow-gpu

    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第9张图片

    -测试tensorflow

    命令行输入python

    输入以下代码:

    import tensorflow as tf

    hello=tf.constant('Hello')

    sess=tf.Session()

    print(sess.run(hello))

    结果输出‘Hello’即为安装成功

    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第10张图片

  5. 安装keras

    参考链接:https://blog.csdn.net/qq_33856151/article/details/79295086

    -激活tensorflow安装环境

    打开Anaconda Prompt输入: activate tensorflow

    -在线安装keras

           输入:pip install keras
    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第11张图片

    -测试keras

    命令行输入python

    输入以下代码:

    Import keras

    显示Using TensorFlow backend

    再输入以下代码:

    keras.__version__

    显示keras的版本号 '2.2.4'
    深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南_第12张图片

你可能感兴趣的:(深度学习环境搭建及适配(WIN10+Python+TensorFlow-gpu+Keras+CUDA+cuDNN)——新手指南)