Windows下PyTorch開發(fā)環(huán)境安裝教程
Anaconda安裝
Anaconda是為方便使用python而建立的一個(gè)軟件包,其包含常用的250多個(gè)工具包,多版本python解釋器和強(qiáng)大的虛擬環(huán)境管理工具,所以Anaconda得名python全家桶。Anaconda可以使安裝、運(yùn)行和升級(jí)環(huán)境變得更簡(jiǎn)單,因此推薦安裝使用。
安裝步驟:
- 官網(wǎng)下載安裝包 https://www.anaconda.com/distribution/#download-section
- 運(yùn)行安裝包
- 選擇安裝路徑:通常選擇默認(rèn)路徑,務(wù)必勾選Add Anaconda to the system PATH environment variable(將Anaconda添加到環(huán)境變量中),等待安裝完成
- 驗(yàn)證安裝成功:快捷鍵win+R,打開cmd,輸入conda,回車,如果出現(xiàn)各種相關(guān)信息,說(shuō)明安裝成功。
PyCharm安裝
PyCharm是強(qiáng)大的 Python IDE,擁有調(diào)試、語(yǔ)法高亮、Project管理、代碼跳轉(zhuǎn)、智能提示、版本控制等功能。
安裝步驟:
- 官網(wǎng)下載安裝包 https://www.jetbrains.com/pycharm/,安裝包分為專業(yè)版(收費(fèi))和社區(qū)版(免費(fèi))。
- 運(yùn)行安裝包。
- 選擇路徑,勾選Add launchers dir to the PATH,勾選.py,等待安裝完成。
CUDA與CuDNN安裝(非必須)
- 檢查是否有合適GPU,若有,需安裝CUDA與CuDNN。只有N卡支持cuda,如下操作可以查看支持的cuda版本:
- NVIDIA控制面板→系統(tǒng)信息→組件→3D設(shè)置/NVCUDA.DLL
- 進(jìn)入PyTorch官網(wǎng)https://pytorch.org/,點(diǎn)擊GetStarted,查看所支持的CUDA版本是多少。
- 進(jìn)入CUDA官網(wǎng)https://developer.nvidia.com/cuda-toolkit-archive,選擇相應(yīng)版本的CUDA,選擇相應(yīng)的操作系統(tǒng),Installer Type 選擇 local。點(diǎn)擊下載第一個(gè)文件。運(yùn)行安裝包,安裝完成不必創(chuàng)建快捷方式。
- 驗(yàn)證CUDA是否安裝成功:進(jìn)入安裝路徑的bin文件夾,復(fù)制路徑,命令行切換到該路徑下(如cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin),然后執(zhí)行nvcc -V,回車,如果出現(xiàn)相關(guān)版本信息,說(shuō)明正確安裝。
- 進(jìn)入cuDNN官網(wǎng)https://developer.nvidia.com/rdp/cudnn-download,注冊(cè)并登錄賬號(hào),選擇相應(yīng)版本下載。解壓安裝包,將里面的三個(gè)文件夾復(fù)制到CUDA安裝路徑下(如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1),cuDNN就安裝完成了。
- 驗(yàn)證cuDNN是否安裝成功:命令行切換到安裝路徑的extras\demo_suite文件夾下,執(zhí)行bandwidthTest.exe,回車,Result = PASS說(shuō)明安裝成功。繼續(xù)執(zhí)行deviceQuery.exe,回車,顯示GPU型號(hào),Result = PASS,表示CUDA和cuDNN都安裝成功了。
將pip源更換到國(guó)內(nèi)鏡像
使用pip安裝python工具包時(shí),由于默認(rèn)源網(wǎng)速極慢,建議切換國(guó)內(nèi)鏡像。
默認(rèn)源
https://pypi.org/
國(guó)內(nèi)鏡像
阿里云 https://mirrors.aliyun.com/pypi/simple/
中國(guó)科技大學(xué) https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) https://pypi.douban.com/simple/
清華大學(xué) https://pypi.tuna.tsinghua.edu.cn/simple/
華中科技大學(xué)https://pypi.hustunique.com/
臨時(shí)使用
安裝命令后接 -i url 即可,如:用清華鏡像下載pandas
pip install pandas -i https://pypi.tuna.tsinghua.edu.cn/simple
永久修改
修改%HOMEPATH%\pip\pip.ini文件
%HOMEPATH%通常為C:\Users\xx
通常需要新建文件夾pip,然后新建文件pip.ini,ini文件可以用記事本打開和編輯,在pip.ini文件中輸入以下內(nèi)容(以豆瓣鏡像為例):
[global]
index-url = https://pypi.douban.com/simple
[install]
trusted-host = pypi.douban.com
PyTorch安裝
1. 下載whl文件(非必須)
進(jìn)入PyTorch官網(wǎng)https://pytorch.org/,點(diǎn)擊GetStarted,選擇相應(yīng)版本,package選擇pip,可以看到下方出現(xiàn)了torch和torchvision的最新版本號(hào),以及一個(gè)網(wǎng)址https://download.pytorch.org/whl/torch_stable.html,復(fù)制網(wǎng)址,進(jìn)入,可以看見各個(gè)版本的torch安裝文件,通過(guò)這種方式,比直接執(zhí)行安裝命令速度要快。文件命名是有規(guī)律的,如:
cu101/torch-1.4.0-cp37-cp37m-win_amd64.whl,
cu101/torchvision-0.5.0-cp37-cp37m-win_amd64.whl
cu后為cuda版本,torch后為torch版本,torchvision后為torchvision版本,cp后為python版本,最后面代表windows64位。
可以利用快捷鍵ctrl+F搜索最新版本的torch和torchvision文件(如搜索cu101/torch-1.4.0),再選擇相應(yīng)python版本和平臺(tái)下載pytorch與torchvision的whl文件,python版本要與系統(tǒng)一致,可以在命令行輸入python來(lái)查看系統(tǒng)中python的版本。
2. 用PyCharm新建一個(gè)項(xiàng)目
Create New Project→Pure Python→命名→Create
新建一個(gè)腳本:File→New→Python file→命名→回車
在腳本中輸入如下代碼→右鍵→Run ‘項(xiàng)目名" →報(bào)錯(cuò)找不到torch,因?yàn)樵诋?dāng)前環(huán)境中,我們沒(méi)有安裝PyTorch。
import torch print("hello pytorch{}".format(torch.__version__)) print(torch.cuda.is_available())
3. 創(chuàng)建python虛擬環(huán)境
點(diǎn)擊下方Terminal→輸入conda create -n 虛擬環(huán)境名 python=版本號(hào)(如conda create -n pytorch_gpu python=3.7)→回車→等待完成
進(jìn)入虛擬環(huán)境:輸入conda activate 虛擬環(huán)境名→回車
4. 安裝
進(jìn)入whl文件所在目錄:輸入cd whl文件所在目錄→回車
安裝:輸入pip install torch→按tab鍵自動(dòng)補(bǔ)全→回車→等待成功安裝
輸入pip install torchvision→按tab鍵自動(dòng)補(bǔ)全→回車→等待成功安裝
注意: 如果第一步?jīng)]有下載whl文件,那么直接用pip或conda命令安裝,安裝命令在PyTorch官網(wǎng)選擇相應(yīng)版本后會(huì)顯示。
5. 將當(dāng)前項(xiàng)目關(guān)聯(lián)到新創(chuàng)建的虛擬環(huán)境,即選擇python解釋器
File→Setting→Project:項(xiàng)目名/Project Interpreter→設(shè)置按鈕→Add→Conda Environment→Existing environment→interpreter中選擇 anaconda安裝路徑/envs/虛擬環(huán)境名/python.exe→OK→OK→OK→稍等片刻進(jìn)行初始化
6. 驗(yàn)證
右鍵運(yùn)行,成功輸出PyTorch版本。如果輸出True,證明GPU可用。
總結(jié)
到此這篇關(guān)于Windows下PyTorch開發(fā)環(huán)境安裝教程的文章就介紹到這了,更多相關(guān)windows PyTorch環(huán)境安裝內(nèi)容請(qǐng)搜索以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持!
相關(guān)文章:
1. Windows Server 2016+Sql Server 2016搭建AlwaysOn集群(一)2. Windows服務(wù)器默認(rèn)IE瀏覽器無(wú)法下載文件的解決方法3. 關(guān)于Windows Server 2012上安裝.NET Framework 3.5的問(wèn)題4. Windows Server 2019 DNS服務(wù)器配置方法(入門篇)5. Windows server 2022創(chuàng)建創(chuàng)建林、域樹、子域的步驟6. Windows server 2003卸載和安裝IIS的圖文教程7. Windows Server 2019和Windows Server, Version 1909的區(qū)別是什么8. Windows Server 2022 超融合部署(圖文教程)9. windows下FTP匿名登錄或弱口令漏洞及服務(wù)加固10. Windows server 2016服務(wù)器基本設(shè)置
