有声小说下载,大主宰之灵路天蚕土豆

主頁 > 知識(shí)庫 > Windows下PyTorch開發(fā)環(huán)境安裝教程

Windows下PyTorch開發(fā)環(huán)境安裝教程

Anaconda安裝

Anaconda是為方便使用python而建立的一個(gè)軟件包，其包含常用的250多個(gè)工具包，多版本python解釋器和強(qiáng)大的虛擬環(huán)境管理工具，所以Anaconda得名python全家桶。Anaconda可以使安裝、運(yùn)行和升級(jí)環(huá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)信息，說明安裝成功。

PyCharm安裝

PyCharm是強(qiáng)大的 Python IDE，擁有調(diào)試、語法高亮、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)版本信息，說明正確安裝。
進(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說明安裝成功。繼續(xù)執(zhí)行deviceQuery.exe，回車，顯示GPU型號(hào)，Result = PASS，表示CUDA和cuDNN都安裝成功了。

將pip源更換到國內(nèi)鏡像

使用pip安裝python工具包時(shí)，由于默認(rèn)源網(wǎng)速極慢，建議切換國內(nèi)鏡像。
默認(rèn)源

https://pypi.org/

國內(nèi)鏡像

阿里云 https://mirrors.aliyun.com/pypi/simple/

中國科技大學(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安裝文件，通過這種方式，比直接執(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來查看系統(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)境中，我們沒有安裝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)行初始化

６. 驗(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)文章希望大家以后多多支持腳本之家！

標(biāo)簽：回訪巴彥淖爾德州廊坊紅河滁州廣安湛江

巨人網(wǎng)絡(luò)通訊聲明：本文標(biāo)題《Windows下PyTorch開發(fā)環(huán)境安裝教程》，本文關(guān)鍵詞 Windows,下,PyTorch,開發(fā),環(huán)境,；如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題，煩請(qǐng)?zhí)峁┫嚓P(guān)信息告之我們，我們將及時(shí)溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò)，涉及言論、版權(quán)與本站無關(guān)。