Win10下Pytorch安裝、環(huán)境搭建、及報錯的解決方案

在學(xué)習(xí)Pytorch的時候,對安裝過程中遇到的問題進行總結(jié)。
因為Pytorch可以啟用GPU進行計算,所以需要查看顯卡的版本信息。
基本環(huán)境為:
系統(tǒng):win10
python:3.6.5

Part1:基礎(chǔ)安裝步驟

第一步:
anaconda安裝完成后可以開始創(chuàng)建環(huán)境,這里以win10 系統(tǒng)為例。打開Anaconda Prompt
(1)先找到anaconda的安裝目錄,Scripts目錄下,運行activate.bat命令,會看到前面有個base
利用conda create -n pytorch python=3.6 創(chuàng)建pytorch獨立環(huán)境


圖片0.png

(2)切換到pytorch環(huán)境
activate pytorch
(3)桌面上 右鍵-NVIDIA控制面板,在幫助中點擊系統(tǒng)信息,點擊組件,查看CUDA的版本信息。我的CUDA是10


圖片1.png
圖片2.png

第二步:
去Pytorch官網(wǎng)上 https://pytorch.org/,選擇自己對應(yīng)的版本信息,我選擇的是windows pip 3.6 cuda10,然后點擊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錯誤可以重新試下,或者先下載到本地,再安裝

圖片3.png

第三步:
安裝完后,在python的交互環(huán)境中進行測試,若import troch,沒有報錯,則說明成功;torch.cuda.is_available() 返回的是True,說明GPU版本的pytorch可以使用

圖片4.png

圖片5.png

第四步:
配置 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:錯誤問題解決方法

錯誤一:ModuleNotFoundError:No nodule named 'numpy.core._multiarray_umath'
該錯誤是有Numpy的版本較低導(dǎo)致,只需要將Numpy升級到最新即可。
pip3 install --upgrade numpy

圖片6.png

錯誤二:torch.cuda.is_available()返回的False
解決辦法:先查看系統(tǒng)信息中cuda的版本號和 選擇的cuda版本是否一致,若不一致,需要重新安裝pytorch;或者將cuda版本進行更新,具體更新操作如下:

右鍵-我的電腦-管理-設(shè)備管理器,找到顯示適配器,選擇NVIDIA,右鍵-更新驅(qū)動程序,將更新到最新版本,重啟電腦即可


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

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

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