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

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

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

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

所以按照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這個統(tǒng)一計算設備架構(gòu)，是基于顯卡的。在以前的系列文章里我們已經(jīng)介紹過Ubuntu下安裝。現(xiàn)在說Windows系統(tǒng)，有時候，筆記本電腦沒有獨立顯卡或者獨立顯卡配置太低，可能就不支持CUDA或者沒法用。這個可以在Windows系統(tǒng)里的，設備管理器-顯示適配器里查看。這種情況，可以試試CPU(作死的節(jié)奏)。那么對應命令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)就知道了。另外機智客看官方網(wǎng)站上的說明，CUDA10版本需要Windows10，而很多電腦系統(tǒng)還停留在win7和win8這些。

總結(jié)

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

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