PyTorch兩種安裝方法
本文安裝的是pytorch1.4版本(cpu版本)
首先需要安裝Anaconda
是否需要安裝基于cuda的PyTorch版本呢?
對(duì)于普通筆記本來說即使有顯卡性能也不高,跑不動(dòng)層數(shù)較深的深度學(xué)習(xí)網(wǎng)絡(luò),所以就不用裝cuda啦。實(shí)際應(yīng)用時(shí)深度學(xué)習(xí)肯定離不開基于高性能GPU的cuda,作為一般的筆記本,基本都跑不動(dòng)數(shù)據(jù)量較大的模型,所以安裝CPU版的PyTorch即可。以后如果繼續(xù)進(jìn)行深度學(xué)習(xí)的研究或開發(fā),都會(huì)基于高性能服務(wù)器,此時(shí)安裝PyTorch版本肯定是選擇有cuda的版本了。
然后進(jìn)入PyTorch官網(wǎng)https://pytorch.org/,選擇相應(yīng)版本的pytorch(本文安裝的是1.4版本的),復(fù)制“Run this Command”里面的代碼
然后打開Anaconda Prompt
運(yùn)行剛剛在官網(wǎng)復(fù)制的命令,如下:
conda install pytorch torchvision cpuonly -c pytorch
系統(tǒng)開始進(jìn)行安裝pytorch
等到屏幕上出現(xiàn)Proceed([y]/n)?時(shí),輸入y
接下來就開始自動(dòng)下載安裝了:
測(cè)試一下是否安裝成功:
首先輸入python,回車。接下來繼續(xù)輸入:
//測(cè)試: import torch import torchvision print(torch.__version__)
PyTorch1.4,cpu版本,就安裝成功啦
到此這篇關(guān)于PyTorch兩種安裝方法的文章就介紹到這了,更多相關(guān)PyTorch安裝內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
解決Tensorflow 使用時(shí)cpu編譯不支持警告的問題
今天小編就為大家分享一篇解決Tensorflow 使用時(shí)cpu編譯不支持警告的問題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-02-02tensorflow實(shí)現(xiàn)在函數(shù)中用tf.Print輸出中間值
今天小編就為大家分享一篇tensorflow實(shí)現(xiàn)在函數(shù)中用tf.Print輸出中間值,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-01-01淺談Python從全局與局部變量到裝飾器的相關(guān)知識(shí)
今天給大家?guī)淼氖顷P(guān)于Python的相關(guān)知識(shí),文章圍繞著Python從全局與局部變量到裝飾器的相關(guān)知識(shí)展開,文中有非常詳細(xì)的介紹及代碼示例,需要的朋友可以參考下2021-06-06python進(jìn)行數(shù)據(jù)合并concat/merge
這篇文章主要介紹了python進(jìn)行數(shù)據(jù)合并concat/merge,文章圍繞主題展開詳細(xì)的內(nèi)容介紹,具有一定的參考價(jià)值,感興趣的小伙伴可以參考一下2022-09-09