PyTorch兩種安裝方法

本文安裝的是pytorch1.4版本（cpu版本）

首先需要安裝Anaconda

是否需要安裝基于cuda的PyTorch版本呢？

對于普通筆記本來說即使有顯卡性能也不高，跑不動層數較深的深度學習網絡，所以就不用裝cuda啦。實際應用時深度學習肯定離不開基于高性能GPU的cuda，作為一般的筆記本，基本都跑不動數據量較大的模型，所以安裝CPU版的PyTorch即可。以后如果繼續進行深度學習的研究或開發，都會基于高性能服務器，此時安裝PyTorch版本肯定是選擇有cuda的版本了。

然后進入PyTorch官網https://pytorch.org/，選擇相應版本的pytorch（本文安裝的是1.4版本的），復制“Run this Command”里面的代碼

然后打開Anaconda Prompt

運行剛剛在官網復制的命令，如下：

系統開始進行安裝pytorch

等到屏幕上出現Proceed([y]/n)?時，輸入y

接下來就開始自動下載安裝了：

測試一下是否安裝成功：

首先輸入python，回車。接下來繼續輸入：

PyTorch1.4，cpu版本，就安裝成功啦

到此這篇關于PyTorch兩種安裝方法的文章就介紹到這了