在本文的tensorflow的環境搭建中,所需要提前說明的是,我的安裝配置的tensorflow-gpu的版本是1.10.0,且相應的cuda的版本是,提前說明的原因是tensorflow-gpu的安裝版本和cuda以及cudnn的版本必須對應且正確。以下則是tensorflow-gpu不同版本的對應的cuda9.0和對應的cudnn7.3的說明圖,在這裏只提到了GPU版本的配置,相關的CPU版本的配置就不做說明了,網上也有許多,配置過程也相對簡單一些:
1.所需安裝程序和文件
- CUDA 9.0-windows10-x64版本
- cudnn cudnn-9.0-windows10-x64-7.3.0版本
- 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是否安裝成功
用圖說明,這個比較簡單,就不用文字說明了,看一下下圖吧。
謝謝大家指教,這個是完整的版本安裝和配置,有什麼建議,可以留言。謝謝!