windows下簡單搭建Tensorflow,keras(GPU)環境

   在本文的tensorflow的環境搭建中,所需要提前說明的是,我的安裝配置的tensorflow-gpu的版本是1.10.0,且相應的cuda的版本是,提前說明的原因是tensorflow-gpu的安裝版本和cuda以及cudnn的版本必須對應且正確。以下則是tensorflow-gpu不同版本的對應的cuda9.0和對應的cudnn7.3的說明圖,在這裏只提到了GPU版本的配置,相關的CPU版本的配置就不做說明了,網上也有許多,配置過程也相對簡單一些:

1.所需安裝程序和文件

  1.  CUDA  9.0-windows10-x64版本
  2.  cudnn  cudnn-9.0-windows10-x64-7.3.0版本
  3.  Anaconda3-5.0.1-Windows-x86_64

2.下載相關的包

2.1 下載Anaconda

下載Anaconda3-5.0.1,在Anaconda官網下載或者用其他方式下載Anaconda3-5.0.1,這個下載的anaconda對應的python版本是3.5,現在一般配置tensorflow-gpu對應的python版本是3.5以上,且一般是python3.5和python3.6版本。

安裝anaconda的時候,需要勾選第一項,如下圖所示

其他的安裝步驟則是簡單的下一步,下一步即可完成。

安裝完成的時候,由於勾選的第一項,會在環境變量中有如下圖的顯示,證明anaconda的環境變量是自己配置好的,如下:

2.2 下載CUDA

咱們來下載CUDA啊,移步NVIDIA下載網站:https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal,下載選項看下圖:

2.3 下載cudnn

下載cuda後,在https://developer.nvidia.com/rdp/form/cudnn-download-survey下準備下載cudnn對應的版本,下載時需要註冊登錄一個賬號,,登陸進去後選擇對應版本,如下圖所示

3.相關安裝包和文件的配置

3.1 CUDA的安裝過程

x

在進行到選項這裏的時候,一般要勾選掉Visual Studio,一般纔會順利進行下一步的安裝。

安裝效果如下所示則爲成功,然後重新啓動計算機則進行下一步

3.2 cudnn的下載

下載完後解壓得到三個文件夾如下圖所示:

將三個文件夾下的對應的文件複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0下的對應的bin,include,lib目錄

然後配置環境變量,安裝成功後有:

然後配置path,如下圖所示,有兩個路徑了,還需要添加

這樣環境變量就配置成功了。

4. 在cmd下安裝配置相應的包:

4.1 安裝tensorflow-gpu

pip install -i https://pypi.douban.com/simple/ tensorflow-gpu==1.10
執行此命令,在cmd之中,會下載tensorflow-gpu很快,且不會因爲下載速度太慢而失敗。且記tensorflow-gpu前面是有一個空格的。且tensorflow-gpu必須爲1.10版本。記:等號必須是兩個== 。

4.2 安裝opencv-python

這個是本人學習圖像處理方面的東西需要的包,做深度學習方面的東西也是和圖像處理相關的,所以需要用到opencv。這個不是必須要安裝的

4.3 安裝keras

pip install -i https://pypi.douban.com/simple/ keras

同樣的道理,運用豆瓣源安裝是很快的,且記住有空格,這裏keras的版本就不是很重要了,不是需要注意的事項。

5檢測tensorflow-gpu是否安裝成功

用圖說明,這個比較簡單,就不用文字說明了,看一下下圖吧。

謝謝大家指教,這個是完整的版本安裝和配置,有什麼建議,可以留言。謝謝!

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章