本人电脑安装TensorFlow-GPU为1.5版本。
对于TensorFlow-GPU的安装较为复杂,需要TensorFlow-GPU、cuda、cudnn安装特定版本。
本文安装TensorFlow-gpu版本为1.5、cuda为9、cudnn为7 、anaconda为3.5。
以下为TensorFlow-gpu的安装。
***********打开anaconda prompt 界面,输入pip install tensorflow-gpu==1.5rc1 --user</span>命令(如果anaconda安装在D盘不需要加 --user</span>命令,只当anaconda安装在C盘user文件夹再试加此命令)****************
**********系统自动进行下载相应的包并且进行安装。
**********安装完毕,安装界面会提示安装成功。
**********输入“conda list”命令回车,进行查看已安装的相应的包。
接下来进行安装相应的cuda、cudnn软件并进行验证TensorFlow-gpu的可用性。