最近准备学习机器学习和深度学习,所以入坑Tensorflow,之前一直使用的是Anaconda3的cpu版本的Tensorflow,但是这次作死一直想用GPU版本的,主要是不想浪费我的1080ti,但是没想到让我走上了一条心酸的魔鬼之路。

最开始是想在Vwmare+Ubuntu16.04+Tensorflow+CUDA,但是倒腾了一晚上,均失败告终,总结经验大概是虚拟机没办法调用本地的物理显卡,所以我放弃了,真的心累,耽误时间不说,还心神俱疲。

虚拟环境搭建失败之后,我就准备在本地win10搭建GPU环境,最开始是 win10 + Tensorflow-GPU 1.12.0 + CUDA 10 + Python 3.6.7,但是我装了很久没成功,查询Tensorflow官方文档之后,得知1.12和CUDA10截至目前2018-11-15日不相匹配(也不排除我装不来),具体的报错可以参考:https://tensorflow.google.cn/install/errors   

  我当时就是遇到了: 

ImportError: DLL load failed: The specified module could not be found.
遇到这个错误是因为我没有正确安装 visual studio Integration ,关于 visual studio Integration failed 的问题我建议
在windows安全启动的模式下删除已经安装的所有Nivida 相关文件,包括驱动和CUDA 然后重新安装CUDA(安装CUDA会顺带安装驱动)

环境变量的设置参考:https://tensorflow.google.cn/install/gpu

我将CUDA10.0换成9.0(不是9.2)之后,再卸载vs C++ 2017 然后安装CUDA9.0之后就成功安装。

安装的顺序是 CUDA  -> tensorflow-GPU -> cudnn。具体的详细步骤可以参考其他博文。遇到 vs问题就卸载重装。可以按照我这个版本来安装。

posted on 2018-11-15 22:02  曹战  阅读(3342)  评论(0编辑  收藏  举报