1070Ti+Tensorflow 1.6

去年双十一的时候,入手了老黄的GTX 1070Ti显卡,这也是我人生中买的第一块显卡。

入手之后,本来是打算来做机器学习的,无奈当时新卡CUDA支持有限,只能用起Cuda8.0,当时的环境是在Anaconda上用conda搭建的,tensorflow的版本还是用的1.2.1。搭建完了,也就放在那里了。

正好上周已经完成了语音控制远程播放电影的功能,难得周末有时间来玩玩新东西,就把最近很热的faceswap项目整上来玩一下。

发现目前tensorflow已经在1.5版本支持了CUDA9.0,和CUDNN7.0.

而最新的CUDA已经出到了9.1,CuDNN也更新到了7.1,不免一阵手痒,倒腾起来。

经过半天的折腾发现:

1)目前CUDA9.1是支持1070Ti的,而CUDA9.0安装时会报错不支持

2)目前最新的tensorflow-gpu 1.5/1.6版本默认是配合使用CUDA9.0的

3)Window上有第三方编译的tensorflow-gpu 1.6版本支持CUDA9.1,具体见后面参考链接(感谢fo40225,好人一生平安)

4) 预编译的wheel默认使用的cudnn是7.0.5版本,切莫使用7.1.1版本,否则后面使用会报错。

最后放出各版本组合,供大家参考:

1.Geforce Game Ready Driver 版本:388.19

2.Python版本:3.6.3

3.Cuda版本:9.1.85

4.Cudnn版本:7.0.5

5.tensorflow-gpu wheel版本:cuda91cudnn71avx2

最后放张图看下GPU使用效果:

参考:

1. tensorflow-gpu 1.6 支持CUDA9.1的wheel下载地址:

https://github.com/fo40225/tensorflow-windows-wheel/tree/master/1.6.0/py36/GPU/cuda91cudnn71avx2

2. windows上编译tensorflow不同特性的步骤:

https://github.com/fo40225/tensorflow-windows-wheel/issues/3

  • 发表于:
  • 原文链接http://kuaibao.qq.com/s/20180311G0V7LB00?refer=cp_1026
  • 腾讯「云+社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 yunjia_community@tencent.com 删除。

扫码关注云+社区

领取腾讯云代金券