1. 程式人生 > >tensorflow Gpu版本 windows 下的安裝

tensorflow Gpu版本 windows 下的安裝

.cn zip壓縮 相關 rtu 官網下載 虛擬 ati com files

主要參考 tensorflow官方文檔 https://tensorflow.google.cn/install/pip?lang=python3

基本環境

顯卡: GTX1050ti
操作系統:win10

預先安裝的軟件

0.軟件列表

打開官網的如下鏈接:https://tensorflow.google.cn/install/gpu 即可看到需要預先安裝的軟件。

1.vs2015 專業版 (社區版也可以)安裝

可以到MSDN,我告訴你下載相關鏡像。完成相應安裝即可。

2. cuda的安裝

到nvidia官網下載cuda9.0版本,這裏下載的(local)版本
版本選擇的截圖如下:
技術分享圖片
如果cuda安裝過程中出現 “This graphics driver could not find compatible graphics hardware”
這是因為cuda內部的顯卡驅動比電腦自身的顯卡驅動版本低,可以跳過警告,繼續安裝。
技術分享圖片

3.cuDnn的安裝

這裏的下載需要註冊nvidia賬號,我安裝時的版本選擇如下,下載後的到一個.zip壓縮文件。
技術分享圖片

官方文檔的安裝過程

1.python3的安裝:(Install the Python development environment on your system)

tensorflow官網顯示只支持python3.4 3.5 3.6 所以需要自行安裝滿足要求的pyhton3版本,可以下載如下圖所示的版本,形化安裝比較方便,而且卸載也比較方便,只需要再次運行安裝程序即可完成卸載。安裝時可以將安裝路徑添加到系統變量path也可以安裝完成後,手動添加。安裝路徑最好不要在C:\Program Files\或C:\Program Files(X86)\路徑下,因為安裝路徑含有空格

有可能會造成一些問題。
技術分享圖片
官方文檔中出現的python3 也就是python 只需要把python安裝目錄下的python.exe文件改為python3.exe即可。
檢測相應的軟件是否存在,不存在的話有可能是沒有添加到系統變量(添加到系統變量即可),如果沒有安裝點擊相應的使用pip也很好安裝
例如安裝virtualenv 執行如下命令:pip install virualenv

3.虛擬環境的創建(Create a virtual environment ):

按照官方文檔說明的步驟就好(俺也一樣O_O)

4.tensorflow的安裝(Install the TensorFlow pip package):

按照官方文檔說明的步驟就好(俺還一樣o_O)

tensorflow Gpu版本 windows 下的安裝