Anaconda下成功安装Tensorflow-gpu
环境:win7 Anaconda3 5.1(对应python3.6)Pyhcharm 显卡:NVIDA Geforce 1050
本来想简简单单安装个tensorflow-gpu,却整整折腾了两天,网上教程和方法很多,但是有很多版本问题,甚至有些方法都是错误的,总得来说就是tensorflow+CUDA+cuDNN之间版本需要相互匹配,否则会有各种问题。以下安装过程参考了网上多篇精华教程(注意每一篇教程只取里面的最简单最有用的一部分),亲测有效,最后运行结果如图:
1.安装tensorflow
官方下载更新工具包的速度很慢,所以安装之前添加清华大学TUNA提供的Anaconda仓库镜像,参考教程:
https://www.cnblogs.com/willnote/p/6746499.html
搜索tensorflow安装网上有很多教程,个人感觉官方的是比较靠谱的,以下附上链接
https://www.tensorflow.org/install/install_windows
我使用Anaconda环境(anacondaprompt)安装的Tensorflow-gpu1.8,详细tensorflow安装步骤可以参考:
https://blog.csdn.net/u012052268/article/details/74202439
2.安装VS
安装CUDA之前,先安装VS(需要其VC++的编译环境,所以只装vc++模块就行)
参考教程:
https://blog.csdn.net/lichuan0987/article/details/80699132
3.安装CUDA
Tensorflow-gpu运行需要CUDA的并行计算包,关于tensorflow+CUDA+cuDNN的安装需要特别注意版本的对应关系,亲身测试tensorflow1.8+CUDA9.0+cuDNN7.1(显卡是Geforce 1050)是可行的组合,具体查看组合的可不可行,可以参考:
https://blog.csdn.net/XCCCCZ/article/details/80385448
版本不匹配会出现各种问题,亲测有效!!!
安装CUDA中注意可以去掉VS intergration的安装(在自定义安装中),否则会出现安装失败,具体也可以参考上一篇文章,或者
https://blog.csdn.net/zzpong/article/details/80282814
4.安装cuDNN
安装完后CUDA后,使用cuDNN加速参考:
https://blog.csdn.net/lanchunhui/article/details/54964064