以前是在Cent OS中运行Tensorflow,,经常需要切换操作系统,很不方便,于是决定在Windows 7下安装Tensorflow。
过程还是挺复杂的,需要安装的包括:Visual Studio、Anaconda、CUDA、cuDNN、PyCharm、Tensorflow-gpu等。
Visual Studio 2017、PyCharm以前已经安装好了,跳过。
然后是Anaconda,从网上下载,然后安装就可以了,比较简单。
CUDA、cuDNN、Tensorflow-gpu对版本是有要求的,不能随便下载,如果版本不兼容就会出错。我安装的是CUDA 9.0、cuDNN 7.0、Tensorflow-gpu 1.5.0。
(1)下载CUDA 9.0,并安装;
(2)下载cuDNN 7.0,解压后有bin、include、lib三个文件夹,将里面的文件分别复制到CUDA Development文件夹下相同名称的文件夹下面;
(3)安装Tensorflow-gpu,在命令窗口中输入,会提示安装成功;
pip install tensorflow--gpu==1.5.0
(4)运行PyCharm,新建一个项目,File-->Settings-->Project-->Project Interpreter-->Add-->System Interpreter,将Anoconda安装目录下的python.exe加进去;
(5)运行一个简单的程序,没有出错,表明安装成功!
import tensorflow as tf print("Version : %s" % tf.__version__)
print("Path : %s" % tf.__path__)