Win10 pip安装GPU版本TensorFlow 指南

本人最近准备学习TensorFlow,网上看了很多资料都是使用anaconda安装,而pip安装资料蛮少的,所以我想分享一下自己的安装过程中取得的经验,自己走过的坑,就不要让其他同学再跳进去了。我的显卡只有GeForce940MX,但我查了一下GeForce 系列都可以支持跑GPU版本,所以我毅然走上了安装GPU版本的不归路(别再掉坑了):
安装过程中容易遇到的三个坑 总结如下
1.python版本
(图片截取自官网:https://www.tensorflow.org/install/install_windows)
Win10 pip安装GPU版本TensorFlow 指南_第1张图片
目前python官网已经有3.7的版本,可惜的是,TensorFlow目前无法使用3.7版本。所以请一定按照官网说明下载python3.5python3.6系列版本!

2.驱动准备条件
(图片截取自官网:https://www.tensorflow.org/install/install_windows)
Win10 pip安装GPU版本TensorFlow 指南_第2张图片
请务必下载正确官方指定的NVIDIA软件!红线标注警示,我们要下载的是CUDA9.0cuDNN v7.0版本(这很重要!!!不然安装TensorFlow的时候会失败)
下载完cuDNN V7.0CUDA9.0之后,请将cuDNN v7.0的文件加入到CUDA9.0对应文件夹下。

3.Visual Studio版本
请下载VS2015及以上,本人使用VS2017企业版,亲测可用。

总结完毕,希望同学们少走弯路。

你可能感兴趣的:(TensorFlow)