软件准备:Anaconda, CUDA, cuDNN, VS
CUDA和cuDNN需要适配,查看GPU软件版本支持情况https://tensorflow.google.cn/install/gpu
下载并安装Anaconda,参考下方链接修改python版本,最新的Anaconda默认python3.7,所以需要新建一个支持tensorflow(使用python3.6)的独立环境
http://docs.anaconda.com/anaconda/faq/#how-do-i-get-anaconda-with-python-3-5-or-3-6安装 VS(安装CUDA需要)
https://visualstudio.microsoft.com安装CUDA
解压cuDNN,将其中的bin、include、lib中的文件拷贝到CUDA安装目录下相应的文件夹
-
安装tensorflow
首先更新pip,再安装tensorflow-gpu版本- python -m pip install --upgrade pip
- pip install --ignore-installed --upgrade tensorflow-gpu
tensorflow中文参考文档:https://github.com/jikexueyuanwiki/tensorflow-zh