Windows环境下安装tensorflow,tensorflow-gpu
- 安装gpu版本需要提前安装CUDA,CUDNN
如果电脑没有gpu,安装cpu版本也是可以的,只是速度会慢很多。cpu和gpu版本之间应该是可以兼容的,默认调用的是gpu版本,建议只安装一个比较不会报错。
Tensor flow cpu版本安装
- 安装anaconda(创建一个python的虚拟环境,里面包含有很多完善的库)
- 升级pip
python -m pip install --upgrade pip
显示Successfully字样即安装成功
- pip更换国内源,换清华的,豆瓣的都可以(这是一次性的换源方法)
pip install SomePackage -i http://pypi.douban.com/simple/
- 安装tensorflow
pip install tensorflow
gpu版本安装(确认有gpu)
-
安装anaconda
-
pip升级
-
pip换源(代码同上)
-
安装cuda
cuda8.0版本下载链接下载之后安装,可以自定义路径,要记住这个路径,后面要用到 -
安装cudnn,实际上就是解压之后把对应的文件放在对应的cuda文件夹里面
cudnn的下载需要先在官网注册账号
cudnn官网 下载cuDNN v5,for CUDA 8.0 下载完成后解压,把里面的文件拷到cuda安装路径下的对应文件夹里面就算安装完成
6. 安装tensorflow-gpu 版本很重要,最好安装低版本,兼容性比较好
pip install tensorflow-gpu==1.2.0
显示Successfully字样即安装成功
接下来检验一下有没有安装成功,在命令行输入pip list
找到上面这个就算安装成功啦!