网上很多教你安装tensorflow-gpu的教程,折腾的那叫一个费劲,而且完事了还不一定跑得起来。
当然,也不能怪他们,都是从tensorflow官网的安装教程过来的。
这里给一种最简单的安装方式,支持windows:
(从tensorflow的issue里看到的)
-
安装Python3.6版本的Anacaonda。
历史版本下载地址
其中5.2.0就对应Python3.6。 -
执行下面的命令:
conda install -c aaronzs tensorflow-gpu
conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn
- 验证
进入anaconda python终端
输入import tensorflow
,没报错就是可用了。
(第一次执行sess.run()时,会花一些时间注册好GPU。之后就不需要了)
网友评论