Win10+Python 3.6环境下cuda 9.1+cuDNN 7.1+Tensorflow 1.7+keras安装

标签: python tensorflow keras
31人阅读 评论(0) 收藏 举报
分类:

安装环境:

  • windows 10 64bit
  • python 3.6

安装以下步骤进行安装:

更新GPU驱动—>安装cuda—>安装cuDNN—>安装Tensorflow—>安装keras

1、更新GPU驱动

首先查看机器的GPU型号,查看其是否支持cuda,在Nvidia官网下载对应的最新驱动进行跟新。这一步应该很简单,就不多说了。

2、安装cuda

Tensorflow已经更新到1.7版本了,官网上说支持最新的cuda 9.X和cuDNN 7.X(结果被坑,后期详述),在Nvidia官网上下载最新的cuda和cuDNN。
cuda 9.1 下载地址:https://developer.nvidia.com/cuda-downloads

cuda9.1

cuDNN 7.1.2 下载地址:https://developer.nvidia.com/rdp/cudnn-download
注意:下载cuDNN需要注册用户,同时下载cuDNN版本时要对应cuda下载的版本,否则运行程序的时候会报错。这里选择v7.1.2 for cuda 9.1的win10版本。

cuDNN 7.1.2

安装包下载好之后,安装cuda(需要管理员权限),按照安装程序一步一步进行下去即可。安装完成后,在cmd输入nvcc -V查看cuda是否安装成功。

nvcc

3、安装cuDNN

解压缩下载的cuDNN安装包,得到以下三个文件夹

cuDNN安装包

将其复制在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1中,并在系统环境变量中添加如下项:

环境变量

其中第一项和第三项是安装cuda时系统自动设置的,我们只需要添加第二项即可。

4、安装tensorflow GPU版本

这一步很简单,在cmd输入 pip install tensorflow-gpu 即可。
输入一下代码,查看是否安装成功

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

结果显示:

ImportError: Could not find 'cudart64_90.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 9.0 from this URL: https://developer.nvidia.com/cuda-toolkit

查了一下,发现tensorflow 1.7这个版本支持到cuda 9.0。由于个人比较懒,不想再次下载安装文件,因此查询了一下其他办法。在https://github.com/fo40225/tensorflow-windows-wheel中有作者利用cuda 9.1自己编译的版本,在项目中找到对应的版本下载就好了(这里下载tensorflow_gpu-1.7.0-cp36-cp36m-win_amd64.whl)。
将之前安装的tensorflow卸载:

pip uninstall tensorflow-gpu

重新安装:

pip install 文件存放路径\文件名.whl

安装完成后,尝试一下代码:

import tensorflow as tf
#Creates a graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
#Creates a session with log_device_placement set to True.
sess = tf.InteractiveSession()
#Runs the op.
print(sess.run(c))

from tensorflow.python.client import device_lib as _device_lib
print(any((x.device_type == 'GPU') for x in _device_lib.list_local_devices()))
print(_device_lib.list_local_devices())

如果能顺利通过,并显示如下结果,说明安装成功。

结果

4、安装keras

输入

pip install keras

keras

安装完成后,在python IDE中输入

import keras

如果输出

keras2

恭喜,keras安装成功,并以tensorflow作为后端。

参考资料
1、https://blog.csdn.net/chai_zheng/article/details/78679881
2、https://blog.csdn.net/weixin_35653315/article/details/71403386
3、https://blog.csdn.net/vcvycy/article/details/79298703

查看评论

tensorflow与cudnn在后面使用中的升级和版本搭配

上次分享了安装经验,安装成功的深度学习环境为,tensorflow1.0.0+cuda8.0+cudnn5.1. 但是随着学习的进度,发现很多其他作者给的源代码都 是用的高版本的tensorflo...
  • CV_YOU
  • CV_YOU
  • 2018-04-11 16:29:10
  • 54

CentOS cuda/tensorflow/keras安装笔记

CentOS cuda/tensorflow/keras安装笔记
  • you12345678901234567
  • you12345678901234567
  • 2017-01-19 21:13:55
  • 554

Windows下安装tensorflow GPU版本报错:OSError: [WinError 126] 找不到指定的模块/Could not find 'cudart64_90.dll'.

       之前一直在学深度学习的理论知识,今日心血来潮,突然想试试深度学习框架的使用,因为之前在学习卷积神经网络时,一直使用的都是python,于是此次选择深度学习框架时,我锁定了TensorFl...
  • wobeatit
  • wobeatit
  • 2018-01-30 16:03:53
  • 9023

OSError: [WinError 126] 找不到指定的模块/Could not find 'cudart64_90.dll'.

OSError: [WinError 126] 找不到指定的模块/Could not find 'cudart64_90.dll'.原文:http://blog.csdn.net/wobeatit/a...
  • jacke121
  • jacke121
  • 2018-02-11 17:30:09
  • 2163

Win10 Anaconda 安装tensorflow-gpu遇到的问题以及解决方案

对于window10安装tensorflow,tensorflow官网已经有比较好的教程网址为:https://tensorflow.google.cn/install/install_windows...
  • qq_31589695
  • qq_31589695
  • 2018-03-26 20:59:49
  • 99

系统环境: windows10 + Anaconda3 + Pycharm

准备:系统环境: windows10 + Anaconda3 + Pycharm(1)环境配置:打开Anaconda Prompt,输入清华仓库镜像,这样更新会快一些: 输入:conda config...
  • nuc_baixu
  • nuc_baixu
  • 2018-03-26 11:11:33
  • 93

Win10下Anaconda3上安装tensorflow教程

                                Win10下Anaconda3上安装tensorflow教程大概花费了4-5个小时的时间安装上了基于GPU版本的tensorflow1....
  • PNAN222
  • PNAN222
  • 2018-03-05 19:54:02
  • 630

win10+python3.5+tensorflow

安装的环境:win10+python3.5+tensorflow在win系统下,安装tensorflow,python版本最低需要3.51.安装Anaconda直接默认安装就行,安装包下载网址:htt...
  • Draw_1999
  • Draw_1999
  • 2018-03-29 10:49:48
  • 48

NVIDIA GEFORCE 720M 装tensorflow-gpu版本

破卡,勿嘲,很早的机子了看自己的卡能不能用,英伟达的查:https://developer.nvidia.com/cuda-gpus其余显卡不知1、安装GPU版本tfpip3 install --up...
  • hiudawn
  • hiudawn
  • 2018-03-29 09:36:56
  • 51

TensorFlow时碰到的问题(GPU)

测试环境:GTX965M,python3.6.x(GPU安装TensorFlow),win10 下面列出了我安装TensorFlow时碰到的问题 1.在安装时候易出现decode错误是由于...
  • mm644706215
  • mm644706215
  • 2017-12-13 22:48:51
  • 4087
    个人资料
    等级:
    访问量: 95
    积分: 50
    排名: 190万+
    文章存档