Windows下深度學習環境快速搭建

---

title: Windows下深度學習環境快速搭建 date: 2019-08-06 10:52:31 tags:

• Deep Learning
• PyTorch
• Tensorflow

---

搞深度學習缺乏不了PyTorch和Tensorflow（固然還有其餘工具），而且有時候還要用到Scipy等工具，因此通過屢次嘗試，發現一條最快的安裝途徑：

本人電腦配置：

• 系統：Windows 10
• 顯卡：NVIDIA GeForce GTX 1050 Ti
• cuda版本：9.0
• cuDNN版本：7.0.5

Python環境配置表：

• Anaconda3-5.2.0-Windows-x86_64.exe
• torch-1.1.0-cp36-cp36m-win_amd64.whl
• torchvision-0.3.0-cp36-cp36m-win_amd64.whl
• tensorflow-gpu==1.12.0

Anaconda的安裝

首先下載anaconda，這個是最佳的解決包衝突的工具，而且安裝了anaconda，順帶着安裝了以後可能會用到的大部分包。

這裏須要注意，anaconda的官網只剩下python2.7個python3.7版本的anaconda了，py27確定不會選擇了，可是py37仍是有不少包不支持，因此仍是安裝py36對應的anaconda比較好。Py36的安裝包能夠在清華的鏡像網站tuna找到。

使用鏡像網站下載，10分鐘左右就能夠下載完成。

PyTorch的安裝

PyTorch官網已經提供了不少安裝方法，conda和pip均可以。不過實際安裝過程當中會遇到網速限制，安裝PyTorch的GPU版本可能會等到死，因此建議先將安裝包下載到本地，而後用pip或者conda執行本地安裝腳本。

下載到本地最方便的方式不是使用連接粘貼到瀏覽器，而是使用一些P2P的下載工具好比迅雷，使用這些工具基本上幾分鐘就能下載好了。

pip的本地安裝指令是pip install [xxx].whl； conda的本地安裝指令是conda install --use-local [xxx].tar.bz2。

安裝完以後記得測試一下是否真正可使用，有時候會遇到一些小問題，可是這些問題均可以經過搜索引擎解決。

好比我曾遇到過的由於numpy版本太低致使PyTorch找不到矩陣乘法的api，而我找到的答案是更新一下numpy便可。

Tensorflow的安裝

tensorflow的版本選擇也是一個坑，當版本大於等於1.13時，只適配cuda10！個人顯卡是1050Ti，只能使用cuda90，所以最終選用tensorflow==1.12.0。

tensorflow的安裝命令比較簡單，TF官網給出的命令是：pip install tensorflow==[版本號]，若是是GPU版本就是pip install tensorflow-gpu==[版本號]。可是若是直接使用pip下載，時間仍是好久的。

最好的方法是先在cmd中執行改指令，而後命令行會提示當前下載的連接，如今就可Ctrl + C結束pip下載了！複製連接到迅雷下載更快！

以後的過程和安裝PyTorch相似。可能會遇到一些小問題，可是均可以完美解決。

Jupyter Notebook的插件安裝

Anaconda自帶了Jupyter Notebook，可是十分很差使。下載Jupyter Notebook插件開啓新世界。

下載插件指令：conda install -c conda-forge jupyter_contrib_nbextensions

具體作法能夠問搜索引擎。