開篇提示，玩機(jī)器學(xué)習(xí)，一定不能用配置一般的電腦，否則后果自負(fù)。現(xiàn)在入正題。其實(shí)不管是在Windows系統(tǒng)還是Ubuntu這樣的Linux系統(tǒng)，要安裝基于Python編程語言的機(jī)器學(xué)習(xí)庫(像PyTorch這樣的機(jī)器學(xué)習(xí)庫)或軟件工具，一般建議都是需要在pip3的格式下安裝的。官方就這么建議和要求的。

之前的文章里，我們已經(jīng)學(xué)習(xí)了如何安裝pip3這樣的工具。Windows和Ubuntu這樣的Linux系統(tǒng)里都已安裝過。那么我們就可以用pip3來安裝庫了。很多時(shí)候，我們要用pip工具來安裝軟件庫，一般直接用pip3 install XXX就行了。所以我們要安裝學(xué)習(xí)開源的機(jī)器學(xué)習(xí)庫PyTorch的話，那么可能就會(huì)在cmd里或者終端shell里執(zhí)行pip3 install pytorch。不過需要注意的是，這個(gè)不總是可行。

有時(shí)候，由于網(wǎng)速問題或者軟件鏡像源的問題，我們直接在pip3里安裝，很容易失敗。當(dāng)然多次嘗試的話，可能會(huì)下載完成，不過那個(gè)幾率并不大。更重要的是，我們需要去國外官方網(wǎng)站或者根據(jù)官方網(wǎng)站的指導(dǎo)建議來下載安裝。

對于Python的機(jī)器學(xué)習(xí)庫PyTorch，我們?nèi)ス倬W(wǎng)看下就知道了。這個(gè)的確不是直接install。而是有不同的版本和命令。我們使用的操作系統(tǒng)不一樣，計(jì)算平臺(tái)不一樣，而由于是Python的pip3安裝，所以包和編程語言就忽略了。根據(jù)OS和計(jì)算平臺(tái)選不同的就行了。

所以按照pytorch官網(wǎng)的建議，如果我們用的是CUDA，那么根據(jù)CUDA版本來選擇適合自己的安裝方法。比如我們用的是CUDA10.2那么就是pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html命令來安裝。而如果是CUDA11.1那么就是pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html命令來安裝。

需要說明一下，CUDA這個(gè)統(tǒng)一計(jì)算設(shè)備架構(gòu)，是基于顯卡的。在以前的系列文章里我們已經(jīng)介紹過Ubuntu下安裝。現(xiàn)在說Windows系統(tǒng)，有時(shí)候，筆記本電腦沒有獨(dú)立顯卡或者獨(dú)立顯卡配置太低，可能就不支持CUDA或者沒法用。這個(gè)可以在Windows系統(tǒng)里的，設(shè)備管理器-顯示適配器里查看。這種情況，可以試試CPU(作死的節(jié)奏)。那么對應(yīng)命令pip install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html執(zhí)行即可。對照官網(wǎng)就知道了。另外機(jī)智客看官方網(wǎng)站上的說明，CUDA10版本需要Windows10，而很多電腦系統(tǒng)還停留在win7和win8這些。

總結(jié)

以上是生活随笔為你收集整理的pytorch用win还是Linux,如何在Windows系统安装使用机器学习库PyTorch的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。