windows下安装tensorflow方法
说明
电脑配置:
* hp笔记本
* CPU Inter Core i7-5500U
* NVIDIA GeForce GTX 950M
所需环境
* CUDA-9.2
* Cudnn-9.2
* Python-3.6
*tensorflow
安装CUDA和Cudnn
CUDA是NVIDIA推出的运算平台,Cudnn是专门针对Deep Learning框架设计的一套GPU计算加速方案。
先安装CUDA
打开首先先解压:
然后就开始安装了,等待安装成功就可以了
解压cudnn的压缩包里面有三个文件夹
把这三个文件夹复制到你cuda的安装目录下,我的地址是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
这时CUDA和CuDNN就安装好了。
安装Tensorflow环境
打开命令行窗口
方法:文件->打开windows Powershell->以管理员身份打开windows Powershell
输入python,打开Python环境
输入 pip install tensorflow(点击Tab键自动补全命令)自动查找对应的tensorflow版本,耐心等待就好。
如果这块出错的话,应该是python版本和tensorflow版本不对应,我的出错就是因为python是32位的,而tensorflow是64位的,重新安装了64位的pytho就好了。
至此,整个安装过程就结束了。