深度學(xué)習(xí)環(huán)境搭建寶典:pytorch+tensorflow
野蠻智能
發(fā)布時(shí)間:03-0815:32
每天給小編五分鐘,小編用自己的代碼,讓你輕松學(xué)習(xí)人工智能。本文將手把手帶你快速搭建你自己的深度學(xué)習(xí)環(huán)境,然后實(shí)現(xiàn)自己的第一個(gè)深度學(xué)習(xí)程序。野蠻智能,小白也能看懂的人工智能。
如果你對(duì)循環(huán)神經(jīng)網(wǎng)絡(luò)的基本原理還不了解,可以通過(guò)小編的精講深度學(xué)習(xí)RNN三大核心點(diǎn),三分鐘掌握循環(huán)神經(jīng)網(wǎng)絡(luò)進(jìn)行簡(jiǎn)單了解。下面進(jìn)入實(shí)戰(zhàn)環(huán)節(jié)。
Anaconda+pytorch環(huán)境準(zhǔn)備
如果你的電腦帶有GPU,可以先安裝Nvidia驅(qū)動(dòng)
+ cuda +
cudnn,然后再搭建環(huán)境,這樣可以達(dá)到更高的運(yùn)行速度。如果不想使用GPU,學(xué)習(xí)階段也可以使用cpu版本,對(duì)于簡(jiǎn)單的程序用CPU和GPU其實(shí)沒(méi)差別。小編這里為大家提供一個(gè)CPU版本的安裝方法。本方法適用于linux和windows平臺(tái)。
第一步:本文采用流行度最高的搭建方法:Anaconda3 + pytorch。首先去清華鏡像站(百度搜索關(guān)鍵字:清華鏡像)進(jìn)入網(wǎng)站后找到anaconda,然后點(diǎn)擊,進(jìn)去后下載Anaconda3-4.2.0,對(duì)應(yīng)的python版本是3.5.2。下載好之后的安裝就是一路默認(rèn)Next就可以。
第二步:在安裝好Anaconda之后。
???? 小編強(qiáng)烈建議為pytorch建立一個(gè)虛擬環(huán)境,這樣做的好處是方便我們同時(shí)使用多個(gè)深度學(xué)習(xí)框架,每個(gè)框架就對(duì)應(yīng)一個(gè)虛擬環(huán)境,框架之間可以互不干擾。具體步驟:Windows用戶(hù)搜索打開(kāi)Anaconda Prompt,然后輸入:source create -n pytorch python=3.5,pytorch是這個(gè)虛擬環(huán)境的名字,你也可以給它起個(gè)別的名字,Python的版本我們也可以指定為3.6。linux用戶(hù)直接打開(kāi)終端,就可以執(zhí)行上述命令。
?? windows 在命令行
conda install pytorch-cpu torchvision-cpu -c pytorch
The following NEW packages will be INSTALLED:
? ninja? ? ? ? ? ? ? pkgs/main/win-64::ninja-1.9.0-py37h74a9793_0
? pytorch-cpu? ? ? ? pytorch/win-64::pytorch-cpu-1.1.0-py3.7_cpu_1
? torchvision-cpu? ? pytorch/win-64::torchvision-cpu-0.3.0-py37_cuNone_1
The following packages will be UPDATED:
? conda? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 4.7.5-py37_0 --> 4.7.10-py37_0
Proceed ([y]/n)? y
Downloading and Extracting Packag
再
conda install pytorch
Collecting package metadata (current_repodata.json): done
Solving environment: done
WARNING conda.core.package_cache_data:_make_single_record(350): Encountered corrupt package tarball at C:\Users\zongsifeng\Anaconda3\pkgs\pytorch-cpu-1.1.0-py3.7_cpu_1.tar.bz2. Conda has removed it, but you need to re-run conda to download it again.
## Package Plan ##
? environment location: C:\Users\zongsifeng\Anaconda3
? added / updated specs:
? ? - pytorch
The following packages will be downloaded:
? ? package? ? ? ? ? ? ? ? ? ? |? ? ? ? ? ? build
? ? ---------------------------|-----------------
? ? _pytorch_select-1.1.0? ? ? |? ? ? ? ? ? ? cpu? ? ? ? ? 2 KB
? ? pytorch-1.0.1? ? ? ? ? ? ? |cpu_py37h39a92a0_0? ? ? ? 23.2 MB
? ? ------------------------------------------------------------
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? Total:? ? ? ? 23.2 MB
The following NEW packages will be INSTALLED:
? _pytorch_select? ? pkgs/main/win-64::_pytorch_select-1.1.0-cpu
? ninja? ? ? ? ? ? ? pkgs/main/win-64::ninja-1.9.0-py37h74a9793_0
? pytorch? ? ? ? ? ? pkgs/main/win-64::pytorch-1.0.1-cpu_py37h39a92a0_0
The following packages will be UPDATED:
? conda? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 4.7.5-py37_0 --> 4.7.10-py37_0
Proceed ([y]/n)? y
Downloading and Extracting Packages
pytorch-1.0.1? ? ? ? | 23.2 MB? | ############################################################################ | 100%
_pytorch_select-1.1. | 2 KB? ? ? | ############################################################################ | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
第三步:然后,打開(kāi)虛擬環(huán)境的方法是,命令行輸入:source activate pytorch,這個(gè)就會(huì)激活虛擬環(huán)境,接下來(lái)就可以在虛擬環(huán)境里工作了,.激活界面如下圖所示:
對(duì)應(yīng)的關(guān)閉虛擬環(huán)境的命令是:source deactivate pytorch,執(zhí)行結(jié)果如下:
第四步,就是在虛擬環(huán)境下,執(zhí)行安裝pytorch框架,具體命令如下
conda install pytorch-cpu torchvision-cpu -c pytorch(因?yàn)槲覀冄b的是cpu版本的,所以windows和linux都是這個(gè)命令哦),這條命令的來(lái)源是pytorch官網(wǎng),如下圖所示。
執(zhí)行的結(jié)果就像這樣:
因?yàn)樾【幰呀?jīng)安裝過(guò)了,所以這里并沒(méi)有下載一大堆包。
最后一步,繼續(xù)在剛才的命令行里執(zhí)行如下命令:conda install nb_conda。這個(gè)程序的功能是使我們能夠使用pytorch虛擬環(huán)境。下面就可以開(kāi)始寫(xiě)自己的測(cè)試程序了。
tensorflow安裝
tensorflow安裝步驟和pytorch的安裝基本是一樣的,第一步建立虛擬環(huán)境,不過(guò)這里可以命名為tensorflow,然后激活虛擬環(huán)境,在虛擬環(huán)境下pip install tensorflow,然后就安裝成功了。代碼測(cè)試與實(shí)戰(zhàn)
剛才的命令行先不要關(guān),我們?cè)佥斎胍粋€(gè)命令:jupyter notebook。
這個(gè)命令的作用是啟動(dòng)notebook,我們將會(huì)在這個(gè)程序里寫(xiě)代碼。它的工作界面就是你的瀏覽器,程序執(zhí)行大概十秒之后會(huì)自動(dòng)跳出瀏覽器界面。如下:
我們點(diǎn)擊右上角的new,就會(huì)找到一個(gè)叫pytorch的工作環(huán)境,這就是我們剛剛建立的虛擬環(huán)境。
點(diǎn)擊這個(gè)pytorch環(huán)境建立一個(gè)新文件,我們就可以開(kāi)始寫(xiě)程序了。
輸入import torch,執(zhí)行(執(zhí)行的方式是shift+Enter),如果不報(bào)錯(cuò),就說(shuō)明我們的安裝完成了。
下面寫(xiě)一個(gè)測(cè)試的程序:
import torcha = torch.Tensor([[2,3],[4,8],[7,9]])print("我定義了一個(gè)矩陣:{}".format(a))
執(zhí)行結(jié)果如下:
大功告成!你已經(jīng)踏入了深度學(xué)習(xí)工程師的大門(mén)!哪里有問(wèn)題的可以在評(píng)論區(qū)留言,如果想繼續(xù)學(xué)習(xí)人工智能,歡迎關(guān)注。我會(huì)每天更新技術(shù)文。野蠻智能,小白也能看懂的人工智能。