1. 程式人生 > 實用技巧 >Anaconda Tensorflow GPU 配置

Anaconda Tensorflow GPU 配置

工具和環境

前言

因為前面配置Visual Studio平臺下Tensorflow C++Api呼叫,已經安裝好Anaconda,並且新增路徑
C:\yourInstallPlace\anaconda3
C:\yourInstallPlace\anaconda3\Scripts
C:\yourInstallPlace\anaconda3\Library\bin
並且已經安裝使用了Tensorflow(不帶cpu或gpu字樣即預設cpu),發現訓練模型很慢,原來還需要特別安裝GPU版的tensorflow,所以開始重新配置。注意:開始前,最好先按照下面

參考博文一測試下自己電腦的顯示卡,我的是兩張Nvidia的GTX1080系列。

安裝TensorFlow-GPU

如果你不會使用Anaconda的切換環境功能,並且有先安裝過tensorflow的最好先解除安裝(此處要選擇你安裝對應的spyder,如果不知道,先谷歌或百度下Anaconda環境切換,spyder,就會明白可以換個環境之間安裝)

pip uninstall tensorflow
pip list
conda uninstall tensorflow
conda list

都嘗試一次,然後呼叫list看看有沒有解除安裝乾淨。解除安裝以後,直接

conda install tensorflow_gpu==1.5//為了和之前安裝好的cuda9.0和cudnn9.0配合使用,直接conda install tensorflow_gpu會給你分配一個不合適的可能,如果你不是cuda9.0和cudnn9.0,請先查詢相應的tensorflow_gpu

測試

如果你前面步驟都成功了,需要先測試下到底能不能使用。還是上面的命令列,輸入:

python
import tensorflow

如果成功後再進入IDE,我的是spyder(此處要選擇你安裝對應的spyder,如果不知道,先谷歌或百度下Anaconda環境切換,spyderIDE),輸入

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

毫無反應就是失敗,有GPU字樣就是成功

[參考博文一:Win10 Anaconda下TensorFlow-GPU環境搭建詳細教程(包含CUDA+cuDNN安裝過程)][anchor-id]
[anchor-id]:

https://www.cnblogs.com/guoyaohua/p/9265268.html

[參考博文二:Wind10環境,Visual Studio2017下用Bazel編譯Tensorflow(失敗轉呼叫編譯好的包配置使用Tensorflowc++)][anchor-id]
[anchor-id]: https://allentdan.github.io/2019/05/08/My-New-Test/

轉載著名:https://allentdan.github.io/2019/05/10/Anaconda-Tensorflow-gpu配置