在Windows系統(tǒng)安裝使用機器學習庫PyTorch

開篇提示,玩機器學習,一定不能用配置一般的電腦,否則后果自負?,F(xiàn)在入正題。其實不管是在Windows系統(tǒng)還是Ubuntu這樣的Linux系統(tǒng),要安裝基于Python編程語言的機器學習庫(像PyTorch這樣的機器學習庫)或軟件工具,一般建議都是需要在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)一計算設(shè)備架構(gòu),是基于顯卡的。在以前的系列文章里我們已經(jīng)介紹過Ubuntu下安裝。現(xiàn)在說Windows系統(tǒng),有時候,筆記本電腦沒有獨立顯卡或者獨立顯卡配置太低,可能就不支持CUDA或者沒法用。這個可以在Windows系統(tǒng)里的,設(shè)備管理器-顯示適配器里查看。這種情況,可以試試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這些。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容