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

Anaconda安裝

  Anaconda是為方便使用python而建立的一個軟件包，其包含常用的250多個工具包，多版本python解釋器和強(qiáng)大的虛擬環(huán)境管理工具，所以Anaconda得名python全家桶。Anaconda可以使安裝、運行和升級環(huán)境變得更簡單，因此推薦安裝使用。

安裝步驟：

• 官網(wǎng)下載安裝包 https://www.anaconda.com/distribution/#download-section
• 運行安裝包
• 選擇安裝路徑：通常選擇默認(rèn)路徑，務(wù)必勾選Add Anaconda to the system PATH environment variable（將Anaconda添加到環(huán)境變量中），等待安裝完成
• 驗證安裝成功：快捷鍵win+R，打開cmd，輸入conda，回車，如果出現(xiàn)各種相關(guān)信息，說明安裝成功。

PyCharm安裝

  PyCharm是強(qiáng)大的 Python IDE，擁有調(diào)試、語法高亮、Project管理、代碼跳轉(zhuǎn)、智能提示、版本控制等功能。

安裝步驟：

1. 官網(wǎng)下載安裝包 https://www.jetbrains.com/pycharm/，安裝包分為專業(yè)版（收費）和社區(qū)版（免費）。
2. 運行安裝包。
3. 選擇路徑，勾選Add launchers dir to the PATH，勾選.py，等待安裝完成。
   

CUDA與CuDNN安裝（非必須）

1. 檢查是否有合適GPU，若有，需安裝CUDA與CuDNN。只有N卡支持cuda，如下操作可以查看支持的cuda版本：
2. NVIDIA控制面板→系統(tǒng)信息→組件→3D設(shè)置/NVCUDA.DLL
3. 進(jìn)入PyTorch官網(wǎng)https://pytorch.org/，點擊GetStarted，查看所支持的CUDA版本是多少。
4. 進(jìn)入CUDA官網(wǎng)https://developer.nvidia.com/cuda-toolkit-archive，選擇相應(yīng)版本的CUDA，選擇相應(yīng)的操作系統(tǒng)，Installer Type 選擇 local。點擊下載第一個文件。運行安裝包，安裝完成不必創(chuàng)建快捷方式。
5. 驗證CUDA是否安裝成功：進(jìn)入安裝路徑的bin文件夾，復(fù)制路徑，命令行切換到該路徑下（如cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin），然后執(zhí)行nvcc -V，回車，如果出現(xiàn)相關(guān)版本信息，說明正確安裝。
6. 進(jìn)入cuDNN官網(wǎng)https://developer.nvidia.com/rdp/cudnn-download，注冊并登錄賬號，選擇相應(yīng)版本下載。解壓安裝包，將里面的三個文件夾復(fù)制到CUDA安裝路徑下（如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1），cuDNN就安裝完成了。
7. 驗證cuDNN是否安裝成功：命令行切換到安裝路徑的extras\demo_suite文件夾下，執(zhí)行bandwidthTest.exe，回車，Result = PASS說明安裝成功。繼續(xù)執(zhí)行deviceQuery.exe，回車，顯示GPU型號，Result = PASS，表示CUDA和cuDNN都安裝成功了。
   

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

  使用pip安裝python工具包時，由于默認(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/

臨時使用
　　安裝命令后接 -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/，點擊GetStarted，選擇相應(yīng)版本，package選擇pip，可以看到下方出現(xiàn)了torch和torchvision的最新版本號，以及一個網(wǎng)址https://download.pytorch.org/whl/torch_stable.html，復(fù)制網(wǎng)址，進(jìn)入，可以看見各個版本的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版本和平臺下載pytorch與torchvision的whl文件，python版本要與系統(tǒng)一致，可以在命令行輸入python來查看系統(tǒng)中python的版本。

2. 用PyCharm新建一個項目

　　Create New Project→Pure Python→命名→Create
　　新建一個腳本：File→New→Python file→命名→回車
　　在腳本中輸入如下代碼→右鍵→Run ‘項目名' →報錯找不到torch，因為在當(dāng)前環(huán)境中，我們沒有安裝PyTorch。

import torch
print("hello pytorch{}".format(torch.__version__))
print(torch.cuda.is_available())

3. 創(chuàng)建python虛擬環(huán)境

　　點擊下方Terminal→輸入conda create -n 虛擬環(huán)境名 python=版本號（如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鍵自動補(bǔ)全→回車→等待成功安裝
　　　　　輸入pip install torchvision→按tab鍵自動補(bǔ)全→回車→等待成功安裝

注意： 如果第一步?jīng)]有下載whl文件，那么直接用pip或conda命令安裝，安裝命令在PyTorch官網(wǎng)選擇相應(yīng)版本后會顯示。

5. 將當(dāng)前項目關(guān)聯(lián)到新創(chuàng)建的虛擬環(huán)境，即選擇python解釋器

　　File→Setting→Project：項目名/Project Interpreter→設(shè)置按鈕→Add→Conda Environment→Existing environment→interpreter中選擇 anaconda安裝路徑/envs/虛擬環(huán)境名/python.exe→OK→OK→OK→稍等片刻進(jìn)行初始化

６. 驗證

　　右鍵運行，成功輸出PyTorch版本。如果輸出True，證明GPU可用。

總結(jié)

到此這篇關(guān)于Windows下PyTorch開發(fā)環(huán)境安裝教程的文章就介紹到這了,更多相關(guān)windows PyTorch環(huán)境安裝內(nèi)容請搜索億速云以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持億速云！