TensorFlow-Gpu 1.8安装

640?wx_fmt=png

TensorFlow-Gpu 1.8安装

0.说在前面

1.Cuda与cuDNN

2.配置及测试

3.tensorflow-gpu 1.8

4.问题

5.作者的话

0.说在前面

终于又到周末了,大家周末快乐,我们都知道在数据挖掘里面有个比赛:Kaggle,这两天在玩Kaggle比赛之泰坦尼克号问题,在下面几节将会详细介绍,泰坦尼克号问题思路及Kaggle比赛玩法。我们一起来期待吧!
这个是当前的成绩:

640?wx_fmt=png

总共有9000多人,排名算了一下在6%到10%之间,线上成绩0.8+。

今天主要分享一下TensorFlow新版gpu版本在win10上安装的问题

版本

  • Win10+PyCharm

  • Cuda9.2

  • cuDNN7.1

  • tensorflow-gpu1.8

点击公众号右下角合作转载->联系我,即可加入我的个人微信,共同探讨交流,以及入交流群(记得备注入群)!

1.Cuda与cuDNN

关于Cuda 9.2与cuDNN 7.1的安装,下载后直接安装即可!
下面是官网链接!

Cuda官网下载:

https://developer.nvidia.com/cuda-toolkit-archive

cuDNN官网下载(需注册):

https://developer.nvidia.com/rdp/cudnn-download

2.配置及测试

关于Cuda 9.2与cuDNN 7.1安装环境配置及测试
【Cuda 9.2配置】

一路安装成功后,dos输入nvcc -V,显示如下信息表示安装成功

640?wx_fmt=jpeg

默认情况

安装好后,默认情况下,系统变量会多出CUDA_PATHCUDA_PATH_V9_2两个环境变量。

添加系统变量

 
 

CUDA_SDK_PATH = C:\ProgramData\NVIDIACorporation\CUDA Samples\v9.2
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

【cuDNN 7.1配置】

解压cudnn-9.2-windows10-x64-v7.1.zip,将文件夹里的内容拷贝到CUDA的安装目录并覆盖相应的文件夹,CUDA默认安装目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

3.tensorflow-gpu 1.8

安装

注:目前tensorflow还不能支持cuda9.2,所以只能通过源码编译或者别人编译好的安装包安装。

下载地址:

https://github.com/fo40225/tensorflow-windows-wheel/tree/master/1.8.0/py36/GPU/cuda92cudnn71sse2

下载这个wheel包后,使用pip install …..whl进行安装

测试

 
 

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

640?wx_fmt=jpeg

4.问题

警告:Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA

 
 

#解决
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

FutureWarning: Conversion of the second argument of issubdtype from float to np.floating is deprecated

 
 

h5py新版本对numpy1.4版本的兼容错误
更新numpy与h5py问题解决

5.作者的话

最后,您如果觉得本公众号对您有帮助,欢迎您多多支持,转发,谢谢!

更多内容,请关注本公众号深度学习系列!

640?wx_fmt=jpeg


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值