Windows下安裝tensorflow-gpu/cpu教程

---

conda安裝GPU版：

---

# 安裝完python后 conda install tensorflow-gpu==1.14 # conda會(huì)自動(dòng)安裝匹配的cuda和cudnn版本

---

pip安裝GPU版：

---

1、安裝cuda

在官網(wǎng)上下載對應(yīng)的版本：https://developer.nvidia.com/cuda-downloads

安裝過程一路next，最好不要修改安裝路徑。正常會(huì)自己添加PATH，不用自己添加

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

2、安裝cuDNN

在官網(wǎng)上下載:https://developer.nvidia.com/rdp/cudnn-download

下載解壓縮后，將文件夾中的cndnn64_7.dll、cudnn.h、cudnn.lib分別復(fù)制到

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版本相對應(yīng)。

3、安裝python

在官網(wǎng)上下載對應(yīng)的安裝包：https://www.python.org
雙擊安裝即可：記得把 Add Python xx to PATH 勾上

4、安裝tensorflow

這里使用pip安裝，python正常已經(jīng)自帶，但是可能需要更新，根據(jù)提示操作即可。

打開cmd，輸入pip3 install tensorflow_gpu

---

PIP安裝CPU版：

---

# 安裝python后 pip3 install tensorflow

---

驗(yàn)證

---

# 驗(yàn)證tf import tensorflow as tf hello = tf.constant('hello!') sess = tf.Session() print(sess.run(hello))# 驗(yàn)證可用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!

總結(jié)

以上是生活随笔為你收集整理的Windows下安装tensorflow-gpu/cpu教程的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。