美文网首页
tensorflow gpu 版本安装(win64)

tensorflow gpu 版本安装(win64)

作者: rogerwu1228 | 来源:发表于2017-07-17 16:23 被阅读121次

软件版本:
1, tensorflow : 1.2.1-gpu

  1. cuda : 8.0.
    3, cudnn: 5.1 for cuda 8.0
    4, python : 3.5
    5, pip 9.0.1

注意事项:
1,在window平台上,只支持python3.5以上版本(推荐使用3.5),使用3.6会有一些坑
2, cuda 和 cudnn 的版本也要对应
3,cudnn这个库建议按照以下顺序加入到对应文件夹:
   cudnn64_5.dll -----------------CUDA\v8.0\bin
   cudnn.h------------------------CUDA\v8.0\include
   cudnn.lib-----------------------CUDA\v8.0\lib\x64

坑: 
https://github.com/tensorflow/tensorflow/issues/7705

相关文章

网友评论

      本文标题:tensorflow gpu 版本安装(win64)

      本文链接:https://www.haomeiwen.com/subject/jwfqkxtx.html