在學(xué)習(xí)Pytorch的時(shí)候,對(duì)安裝過程中遇到的問題進(jìn)行總結(jié)。
因?yàn)镻ytorch可以啟用GPU進(jìn)行計(jì)算,所以需要查看顯卡的版本信息。
基本環(huán)境為:
系統(tǒng):win10
python:3.6.5
Part1:基礎(chǔ)安裝步驟
第一步:
anaconda安裝完成后可以開始創(chuàng)建環(huán)境,這里以win10 系統(tǒng)為例。打開Anaconda Prompt
(1)先找到anaconda的安裝目錄,Scripts目錄下,運(yùn)行activate.bat命令,會(huì)看到前面有個(gè)base
利用conda create -n pytorch python=3.6 創(chuàng)建pytorch獨(dú)立環(huán)境
(2)切換到pytorch環(huán)境
activate pytorch
(3)桌面上 右鍵-NVIDIA控制面板,在幫助中點(diǎn)擊系統(tǒng)信息,點(diǎn)擊組件,查看CUDA的版本信息。我的CUDA是10
第二步:
去Pytorch官網(wǎng)上 https://pytorch.org/,選擇自己對(duì)應(yīng)的版本信息,我選擇的是windows pip 3.6 cuda10,然后點(diǎn)擊Run this Command,復(fù)制
pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-win_amd64.whl
若遇到HTTPERROR錯(cuò)誤可以重新試下,或者先下載到本地,再安裝
第三步:
安裝完后,在python的交互環(huán)境中進(jìn)行測(cè)試,若import troch,沒有報(bào)錯(cuò),則說明成功;torch.cuda.is_available() 返回的是True,說明GPU版本的pytorch可以使用
第四步:
配置 Jupyter Notebook
(1)先安裝ipykernel:conda install ipykernel
(2)加入環(huán)境:python -m ipykernel install --name pytorch --display-name "Pytorch for NLP"
(3)切換回基礎(chǔ)環(huán)境:activate base
(4)創(chuàng)建jupyter notebook配置文件:jupyter notebook --generate-config
part2:錯(cuò)誤問題解決方法
錯(cuò)誤一:ModuleNotFoundError:No nodule named 'numpy.core._multiarray_umath'
該錯(cuò)誤是有Numpy的版本較低導(dǎo)致,只需要將Numpy升級(jí)到最新即可。
pip3 install --upgrade numpy
錯(cuò)誤二:torch.cuda.is_available()返回的False
解決辦法:先查看系統(tǒng)信息中cuda的版本號(hào)和 選擇的cuda版本是否一致,若不一致,需要重新安裝pytorch;或者將cuda版本進(jìn)行更新,具體更新操作如下:
右鍵-我的電腦-管理-設(shè)備管理器,找到顯示適配器,選擇NVIDIA,右鍵-更新驅(qū)動(dòng)程序,將更新到最新版本,重啟電腦即可
|