-----------------2019.7月更新-----------------------

現(xiàn)在安裝tensorflow不需要這么麻煩了，直接安裝anaconda，然后使用命令conda install tensorflow-gpu即可

----------------------------------------------------------

在Windows下安裝Tensorflow有很多需要注意的事項(xiàng)，最近幫不同實(shí)驗(yàn)室的同學(xué)裝了好幾次Tensorflow，其中遇到了很多問(wèn)題，這里也寫個(gè)小的安裝教程，希望能夠幫助大家。

1.電腦配置：

CPU: Intel Core i7-8700

GPU: NVIDIA GTX-1060-6GB (NVIDIA Quadro M4000)

內(nèi)存: 24GB (8GB 2400+16GB 3000)

硬盤: 240G(SSD)+2T(HDD)

IDE: Pycharm

2.安裝準(zhǔn)備：

在安裝Tensorflow-gpu前請(qǐng)記住需要安裝Anaconda、CUDA、cuDNN等組件。

2.1 Anaconda的安裝

Anaconda指的是一個(gè)開源的Python發(fā)行版本，其包含了conda、Python等180多個(gè)科學(xué)包及其依賴項(xiàng)。Anaconda-win10(64位)-python3.6安裝地址。Anaconda的安裝過(guò)程十分簡(jiǎn)單，一直點(diǎn)下一步就好，但是在安裝過(guò)程中記得將其添加到系統(tǒng)的環(huán)境變量中，如下圖所示。

安裝完成后可以到我的電腦-屬性-高級(jí)系統(tǒng)設(shè)置-高級(jí)-環(huán)境變量-Path中查看

安裝完Anaconda之后可以直接通過(guò)Anaconda來(lái)創(chuàng)建編譯環(huán)境，其中包含了tensorflow所依賴的一些包如numpy等，不需要再重復(fù)安裝，這也是為什么用anaconda安裝tensorflow的原因。

2.2 CUDA及cuDNN的安裝

在安裝CUDA和cuDNN之前，記得查看本電腦的顯卡對(duì)應(yīng)什么版本的CUDA及其cuDNN版本，如下圖。

本機(jī)顯卡對(duì)應(yīng)的CUDA可以從桌面右鍵-NVIDIA控制面板-幫助-系統(tǒng)信息-組件中查看：

可以看到本機(jī)GTX1060-6G顯卡對(duì)應(yīng)的是CUDA9.0版本。這個(gè)很重要，因?yàn)橹鞍惭bQuadro M4000系列顯卡時(shí)對(duì)應(yīng)的CUDA8.0版本，我裝的是CUDA9.0所以一直安裝不成功。

可以看到我的電腦對(duì)應(yīng)的是CUDA9.0+cuDNN7.0+Tensorflow-gpu1.5.0-1.10.0。CUDA9.0的安裝地址。CUDA的安裝也是一直下一步即可，安裝完成后會(huì)自動(dòng)將CUDA的相關(guān)文件加入系統(tǒng)環(huán)境變量，如下圖。

cuDNN7.0的安裝地址。在安裝cuDNN的時(shí)候需要注冊(cè)一個(gè)NVIDIA的賬號(hào)才能下載，下載完成后得到是一個(gè)壓縮包，解壓到相應(yīng)的目錄下，在系統(tǒng)環(huán)境變量中加入…\cuda\bin，如下圖：

至此，tensorflow的準(zhǔn)備工作已經(jīng)結(jié)束了，你需要在環(huán)境變量中再檢查一下是否有Anaconda、CUDA以及cuDNN的相關(guān)配置。

3.Tensorflow-gpu的安裝

①首先在命令行窗口中查看Anaconda是否正確安裝。win+R輸入cmd打開命令行窗口，輸入conda –version查看Anaconda是否正確安裝。如果如下圖所示則說(shuō)明Anaconda安裝成功。

②創(chuàng)建python虛擬環(huán)境。在命令行窗口中輸入conda create -name tensorflow-gpu python=3.6。這句的意思是創(chuàng)建一個(gè)名為tensorflow-gpu的python虛擬環(huán)境，python編譯版本為3.6，其中包含了tensorflow必須的一些依賴包，因而不用再額外安裝。（請(qǐng)注意python版本和你所需要安裝的tensorflow對(duì)應(yīng)關(guān)系，部分如下表所示）

安裝過(guò)程如下，按y即可。

等待安裝完成，在anaconda的安裝目錄下的envs文件中出現(xiàn)一個(gè)tensorflow-gpu的子文件夾，我們將在這個(gè)虛擬環(huán)境中完成tensorflow的安裝。(如果找不到anaconda的安裝目錄可以在環(huán)境變量中找到。)

③tensorflow-gpu安裝。激活tensorflow-gpu虛擬環(huán)境，在命令窗口中輸入activate tensorflow-gpu(很重要！)，通過(guò)pip查看已安裝的包，輸入pip list，如下圖：

將pip更新到最新版本，輸入：python -m pip install --upgrade pip

接下來(lái)就是安裝tensorflow-gpu了，輸入pip install tensorflow-gpu，默認(rèn)安裝最新版本，如果需要安裝其他版本，輸入pip install tensorflow-gpu==1.xx.xx。會(huì)提上安裝相應(yīng)的依賴項(xiàng)，按y，等待安裝完成。

④查看tensorflow-gpu是否安裝成功，繼續(xù)在tensorflow-gpu環(huán)境下，輸入pip list查看已安裝的包，如下，安裝成功！

4.程序驗(yàn)證

在pycharm IDE下將相關(guān)編譯器改為tensorflow-gpu環(huán)境下的Python.。點(diǎn)擊File-Setting-Project:xxx-Project Interpreter下的Project Interpreter設(shè)置圖標(biāo)，點(diǎn)擊add local，如下圖所示

等待加載完成后既可以驗(yàn)證tensorflow-gpu是否成功調(diào)用顯卡資源進(jìn)行計(jì)算了。

新建一個(gè)python文件，輸入：

import tensorflow as tfsess = tf.Session() a = tf.constant(2) b = tf.constant(3) print(sess.run(a+b))

若輸出結(jié)果為：

則說(shuō)明tensorflow-gpu正確調(diào)用顯卡資源進(jìn)行計(jì)算。

總結(jié)：

安裝tensorflow-gpu過(guò)程中一定要注意自己的硬件支持什么版本的軟件，已經(jīng)環(huán)境變量的配置是否正確，盡量將不需要的環(huán)境變量刪掉。以后需要編寫深度學(xué)習(xí)程序時(shí)需要其他的包如opencv一律先activate到tensorflow-gpu虛擬環(huán)境中再安裝，就可以直接調(diào)用了。

轉(zhuǎn)載于:https://www.cnblogs.com/Edward-Elric/p/10125893.html

總結(jié)

以上是生活随笔為你收集整理的Tensorflow-gpu在windows系统下的安装及使用(使用Pycharm IDE)的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

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