軟件準(zhǔn)備:Anaconda, CUDA, cuDNN, VS
CUDA和cuDNN需要適配,查看GPU軟件版本支持情況https://tensorflow.google.cn/install/gpu
下載并安裝Anaconda,參考下方鏈接修改python版本,最新的Anaconda默認(rèn)python3.7,所以需要新建一個(gè)支持tensorflow(使用python3.6)的獨(dú)立環(huán)境
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安裝目錄下相應(yīng)的文件夾
-
安裝tensorflow
首先更新pip,再安裝tensorflow-gpu版本- python -m pip install --upgrade pip
- pip install --ignore-installed --upgrade tensorflow-gpu
tensorflow中文參考文檔:https://github.com/jikexueyuanwiki/tensorflow-zh