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)文章
win2003 自動(dòng)注銷已經(jīng)斷開的會(huì)話
win2003 自動(dòng)注銷已經(jīng)斷開的會(huì)話,解決遠(yuǎn)程連接人數(shù)限制的問(wèn)題。2011-01-01IIS服務(wù)器下安裝配置XCache緩存器方法及XCache下載
使用這個(gè)加速器可以很大程度的降低CPU占用(但會(huì)造成更多的內(nèi)存占用),并使網(wǎng)頁(yè)執(zhí)行時(shí)間縮短。具體原理不介紹了,網(wǎng)上有很多2012-01-01win服務(wù)器防止安全策略或防火墻配置錯(cuò)誤而導(dǎo)致遠(yuǎn)程無(wú)法連接的bat
這篇文章主要介紹了win服務(wù)器防止安全策略或防火墻配置錯(cuò)誤而導(dǎo)致遠(yuǎn)程無(wú)法連接的bat,需要的朋友可以參考下2017-04-04windows服務(wù)器中.shtml頁(yè)面訪問(wèn)無(wú)法顯示解決方法
這篇文章主要介紹了windows服務(wù)器中.shtml頁(yè)面訪問(wèn)無(wú)法顯示解決方法,需要的朋友可以參考下2014-11-11圖文詳解本地Windows 7/8上IIS服務(wù)器搭建教程
這篇文章主要以圖文結(jié)合的方式詳細(xì)介紹了本地Windows 7/8上IIS服務(wù)器搭建教程,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2017-04-04IIS 6.0 中修改注冊(cè)表自定義 Httperr.Log
在 Windows 2003 Server 中.IIS 6.0 的日志記錄比 IIS 5.0 要更強(qiáng)大和安全.我們可以通過(guò)設(shè)置日志文件的路徑來(lái)將日志保存在其他地方.比如D盤.因?yàn)槿绻愕?IIS 服務(wù)器流量很大.那放在默認(rèn)的C盤可能會(huì)大量占用硬盤空間2014-01-01windows 10 安裝和使用中5個(gè)常見問(wèn)題
2015年7月29日0點(diǎn)起,Windows 10推送全面開啟,Windows7、Windows8.1用戶可以免費(fèi)升級(jí)到Windows 10,用戶也可以通過(guò)系統(tǒng)升級(jí)到Windows10,在這過(guò)程中,用戶會(huì)遇到這樣那樣的問(wèn)題,下面小編總結(jié)了windows 10 安裝和使用中5個(gè)常見問(wèn)題,需要的朋友可以參考下2015-09-09Windows中利用計(jì)劃任務(wù)實(shí)現(xiàn)自動(dòng)備份文件
我們來(lái)利用windows系統(tǒng)的計(jì)劃任務(wù)來(lái)實(shí)現(xiàn)數(shù)據(jù)庫(kù)的備份,是文件夾copy到固定的文件夾,是把文件夾copy以當(dāng)天日期的自動(dòng)生成的文件夾,大家可以發(fā)揮下結(jié)合下bat效果更佳2013-01-01