win7+cpu+pytorch+深度學(xué)習(xí)環(huán)境

教程鏈接:https://www.freesion.com/article/7398508321/

電腦上裝有anaconda,python3.8, pycharm,這里想要安裝pytorch這一深度學(xué)習(xí)的框架,安裝分GPU和CPU兩個(gè)版本,如果安裝GPU版本,可以先安裝Nvidia驅(qū)動(dòng) + cuda + cudnn,然后再搭建環(huán)境,這樣可以達(dá)到更高的運(yùn)行速度。如果對(duì)運(yùn)行速度沒有太高的要求,學(xué)習(xí)階段也可以使用CPU版本,對(duì)于簡(jiǎn)單的程序用CPU和GPU其實(shí)沒太大差別。

一般來說有兩種安裝方法,一個(gè)是使用conda在線安裝,另一個(gè)是通過pip安裝(需要先從pytorch官網(wǎng)將torch-1.1.0-cp36-cp36m-win_amd64.whl和torchvision-0.3.0-cp36-cp36m-win_amd64.whl下載到本地),這里用conda在線進(jìn)行安裝,但是當(dāng)使用conda安裝的時(shí)候,可能會(huì)遇到下載過慢的問題,這里推薦用清華鏡像源替代默認(rèn)conda源的方法,實(shí)測(cè)速度比默認(rèn)conda源快得多!

下面開始教程。


一、創(chuàng)建虛擬環(huán)境并激活

很多大佬建議為pytorch創(chuàng)建一個(gè)虛擬環(huán)境,這樣做的好處是方便我們可以同時(shí)使用多個(gè)深度學(xué)習(xí)框架,每個(gè)框架對(duì)應(yīng)一個(gè)虛擬環(huán)境,框架之間互不干擾。

打開anaconda的命令行,即anaconda prompt, 輸入以下命令,

conda create -n pytorch-cpu python=3.8

其中pytorch-cpu是虛擬環(huán)境的名字,也可以是其他的。然后會(huì)是這樣:

接下來輸入以下命令激活我們剛剛創(chuàng)建的虛擬環(huán)境

conda activate pytorch-cpu

二、安裝PYTORCH

進(jìn)入pytorch-cpu這一虛擬環(huán)境后,為了加快下載速度,首先輸入以下的命令,添加清華鏡像源,

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

另外為了保險(xiǎn)起見,建議同時(shí)添加第三方conda源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

接下來輸入如下安裝命令,

conda install pytorch torchvision cpuonly

然后會(huì)出現(xiàn)如下界面:

在Proceed([y]/n)? 后面輸入y,回車,即可開始從清華鏡像源下載包,接著等待安裝成功就好

三、測(cè)試環(huán)境是否搭建成功

首先輸入python,可以看到輸出了python的版本信息,

接下來import torch,輸出torch的版本信息,并進(jìn)行測(cè)試,pytorch安裝成功

import torch

print(torch.__version__)

四、配置PYCHARM

打開pycharm,新建一個(gè)項(xiàng)目,添加新的interpreter,找到新建的pytorch-cpu環(huán)境下的python.exe,將其添加到interpreter中,

項(xiàng)目建好后,新建python file,輸入以下代碼進(jìn)行測(cè)試:

import torch

print(torch.__version__)

print(torch.cuda.is_available())

x = torch.randn(4,2)

print(x)

這一步的時(shí)候,報(bào)錯(cuò):

ImportError:

IMPORTANT: PLEASE READ THIS FOR ADVICE ON HOW TO SOLVE THIS ISSUE!

Importing the numpy C-extensions failed. This error can happen formany reasons, often due to issues with your setup or how NumPy wasinstalled.

查了下說是環(huán)境變量的問題,加了(也不知道是哪個(gè)其作用了)

D:\software\anaconda3\envs\pytorch-cpu

D:\software\anaconda3\envs\pytorch-cpu\Scripts

D:\software\anaconda3\envs\pytorch-cpu\Library\bin

加完還不行,重啟下電腦好了。

五、Tips

(1)每次用pytorch時(shí),要先打開Anaconda Prompt,激活我們創(chuàng)建的虛擬環(huán)境pytorch-cpu,使用完畢后執(zhí)行以下命令切換回原來的python環(huán)境,

deactivate pytorch-cpu

(2)如果需要?jiǎng)h除虛擬環(huán)境,則用以下命令來實(shí)現(xiàn),

conda remove -n your_env_name(虛擬環(huán)境名稱) --all

(3)注意使用清華鏡像源時(shí)安裝命令與pytorch官網(wǎng)上給出的命令不同,pytorch官網(wǎng)給的命令是?conda install pytorch torchvision cpuonly -c pytorch

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

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

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