去年双十一的时候,入手了老黄的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
领取专属 10元无门槛券
私享最新 技术干货