win10下安裝tensorflow-gpu，以及將tensorflow-gpu匯入pycharm
win10下安裝tensorflow-gpu，以及將tensorflow-gpu匯入pycharm
一、準備工具
- vs2015 。vs2015我不確定是否要安裝，網上一些教程說要，一些說不要，為了保險起見，我還是裝了。下載地址：官方版
- anaconda。下載地址：[清華映象載]https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
- tensorflow-gpu。
- cuda。
- cudnn。
- pycharm。；
二、安裝過程
-
vs2015
將下載好的壓縮包解壓後，雙擊.exe檔案進行安裝。我們只需要安裝c++元件，見下圖
-
anaconda
雙擊下載好的.exe檔案，一直下一步就可以了，期間你可以自定義安裝路徑，這個路徑以後要用到。
裝好之後，在開始選單就出現了一個Anaconda Prompt，這和cmd特別像。點選Anaconda Prompt，輸入pip install tensorflow-gpu，我這樣裝的是1.12.0的，當然如果你要裝其他版本的，可以使用pip install tensorflow-gpu==你需要的版本號（版本號一定要記住，我們下一步裝cuda和cudnn要用到）。我這裡已經裝好了，所以顯示要求已經滿足。安裝好之後，會有提示Successfully installed tensorflow-gpu。
此時，在Anaconda Prompt中輸入python,然後在輸入import tensorflow as tf會報錯，因為我們還沒有安裝cuda和cudnn
3. cuda和cudnn
為什麼要把這兩個放在一起講呢？因為它們的版本號要相對應，不然會報錯。
上一步我們將tensorflow-gpu裝好後，按照這個地址找到你所安裝的tensorflow的cuda和cudnn版本，然後下載cuda，我這裡下載的是cuda9.0，其他版本的可以直接百度或去英偉達官網下載。cudnn,這裡需要註冊填寫問卷才能下載，我這裡下載的是cuDNN v7.4.2 (Dec 14, 2018), for CUDA 9.0。
（1）下載好cuda之後，雙擊.exe檔案進行安裝。
也許你會遇到上面這種情況，它是說你本機的顯示卡驅動版本比當前要安裝的CUDA自帶的驅動版本高（實際上，不論CUDA裝的哪個版本，只要本機驅動比CUDA自帶驅動版本高，都可能出現這個問題）。
解決辦法：
直接點選繼續—>同意並繼續—>自定義（高階）—>只選擇CUDA進行安裝，最後安裝成功。
在自定義介面可以看到CUDA自帶驅動版本號以及目前本機驅動版本號，如果本機版本號高於CUDA自帶版本號，就不要再勾選安裝了。
（2）驗證CUDA安裝是否成功
開啟cmd，輸入nvcc -V,出現如下類似資訊,說明安裝成功。
（3）在環境變數中確定CUDA_PATH和CUDA_PATH_V_9_0在系統變數中。我這裡之前多裝了一個CUDA8，所以多了一個CUDA_PATH_V_8_0.
(4)解壓下載好的cudnn包，將該包下的所有檔案複製到對應資料夾。預設資料夾在：
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
下面的一步很重要。雙擊上圖中的cuda資料夾，定位到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\cuda\bin資料夾，將該複製地址C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\cuda\bin，然後將該地址新增到環境變數-使用者變數-path中
（5）此時我們再在Anaconda Prompt中輸入python，再輸入import tensorflow as tf,就不會報錯了。
至此，我們的tensorflow-gpu就安裝成功了。
4.將tensorflow-gpu匯入pycharm
（1）pycharm下載、安裝、破解，點選這裡，我是用第三種方法破解的。
（2）匯入。開啟pycharm，點選file-settings-project-project interpreter
點選上面的齒輪圖示,選擇Add,進入如下介面。選擇Existing environment(已經存在的環境)，點選下圖畫圈位置，選擇anaconda下的python.exe就可以了。
至此，所有都完成了。我們在pycharm中使用tensorflow，結果正確。
參考
[1]: https://blog.csdn.net/ygjustgo/article/details/78883981
[2]: https://blog.csdn.net/sb19931201/article/details/53648615
[3]: https://blog.csdn.net/ygjustgo/article/details/78883981