Windows下安装tensorflow-gpu/cpu教程

Windows下安装tensorflow-gpu/cpu教程

---

conda安装GPU版：

---

# 安装完python后
conda install tensorflow-gpu==1.14
# conda会自动安装匹配的cuda和cudnn版本

---

pip安装GPU版：

---

1、安装cuda

在官网上下载对应的版本：https://developer.nvidia.com/cuda-downloads

安装过程一路next，最好不要修改安装路径。正常会自己添加PATH，不用自己添加

PS：tensorflow目前不支持cuda9.1及以上

2、安装cuDNN

在官网上下载:https://developer.nvidia.com/rdp/cudnn-download

下载解压缩后，将文件夹中的cndnn64_7.dll、cudnn.h、cudnn.lib分别复制到

C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin

C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.1\include

C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.1\lib\x64

注意：cudnn需要与cuda版本相对应。

3、安装python

在官网上下载对应的安装包：https://www.python.org
双击安装即可：记得把 Add Python xx to PATH 勾上

4、安装tensorflow

这里使用pip安装，python正常已经自带，但是可能需要更新，根据提示操作即可。

打开cmd，输入pip3 install tensorflow_gpu

---

PIP安装CPU版：

---

# 安装python后
pip3 install tensorflow

---

验证

---

# 验证tf
import tensorflow as tf
hello = tf.constant('hello!')
sess = tf.Session()
print(sess.run(hello))

# 验证可用GPU
import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
print(device_lib.list_local_devices())

---

Good Luck!