Anaconda下成功安装Tensorflow-gpu(精华方法汇总)

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


猜你喜欢

转载自blog.csdn.net/lichuan0987/article/details/80703935