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

在這裡插入圖片描述

運行剛剛在官網復制的命令,如下:

conda install pytorch torchvision cpuonly -c pytorch

系統開始進行安裝pytorch

在這裡插入圖片描述

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

在這裡插入圖片描述

接下來就開始自動下載安裝瞭:

在這裡插入圖片描述

測試一下是否安裝成功:

首先輸入python,回車。接下來繼續輸入:

//測試:
import torch 
import torchvision
print(torch.__version__)

在這裡插入圖片描述

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

到此這篇關於PyTorch兩種安裝方法的文章就介紹到這瞭,更多相關PyTorch安裝內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!