小白秒懂的Windows下搭建基於pytorch的深度學習環境


配置環境總體思路

1.依據python版本選擇對應Anaconda版本;

2.依據顯卡驅動版本選擇對應的CUDA版本;

3.依據CUDA版本選擇對應的cudnn和pytorch版本。

一、Anaconda安裝

1.下載地址

1.官網  https://www.anaconda.com/products/individual

2.清華源  https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

注:若官網下載速度慢,可用清華源下載;注意操作系統的位數(一般是64位的)

2.安裝教程

1.雙擊運行下載的.exe文件,按照以下順序點擊:Next → I Agree

​2.選擇用戶:Just Me 或 All Users 都可以,默認選擇 Just Me

​3.選擇安裝目錄:一般不建議安裝到C盤;可更改安裝路徑,注意安裝路徑不能出現中文、空格,使用純英文即可。

​4.勾選安裝參數:一是添加anaconda到環境變量中(path),二是將anaconda的python版本作為默認的python版本; 建議兩者都勾選,環境變量手動再添加一遍(后續有介紹)。

​5.等待安裝即可,結束時最后 “兩個learn” 都不需要勾選,點擊Finish即可。

3.配置環境變量

1.鼠標選擇”此電腦“,單擊鼠標右鍵,選擇 ”屬性“;

​2.左鍵單擊左側欄的 “高級系統設置”

​3.點擊 “高級”欄中的 “環境變量”

​4.點擊 “系統變量”中的 “Path”項,選擇編輯;

​5.點擊 “新建”,然后添加以下類似路徑(以安裝路徑為准,只需保證最后路徑的文件名稱相同即可)

E:\Anaconda     (Python需要)
E:\Anaconda\Scripts        (conda自帶腳本)
E:\Anaconda\Library\mingw-w64\bin     (使用C with python的時候) 
E:\Anaconda\Library\bin               (jupyter notebook動態庫)
E:\Anaconda\Library\usr\bin  (我自己沒找到此文件夾,若存在可將此條也加上)

  注:切記點擊確定!!

4.安裝測試

1. win+R,輸入cmd,進入命令提示符

2. 輸入:  conda --version

若跳出conda版本,則表示anaconda安裝成功,且已經配置好環境變量

二、CUDA安裝

1.確定版本

控制面板——硬件和聲音——NVIDIA控制面板——幫助——系統信息——組件

1.第三行,NVCUDA64.DLL 可查看顯卡驅動版本支持的最高CUDA版本;

​2.網頁搜索相應CUDA版本安裝,建議不需要安裝太新版本。網頁直接可查找;

2.安裝步驟

​ 1.雙擊運行.exe文件后,前兩項是默認安裝;

​ 2.當選擇安裝選項時,選擇自定義安裝,驅動程序組件第一次安裝盡量全選;

​ 3.記得留意CUDA安裝位置,之后cudnn的安裝會用到!!

3.測試

1.win+R,輸入cmd,進入命令提示符

​2.輸入:nvcc -V

4.添加至系統變量

​ 1. 添加方法與anaconda的相同:**此電腦——屬性——高級系統設置——環境變量——path——編輯——新建**

​ 2.添加以下類似路徑:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp

3.有些路徑在安裝時可能已經添加,手動補充沒有的即可

三、cuDNN配置 

​ 1.進入官網依照cuda版本選擇cudnn文件下載(需要注冊賬號)

​ https://developer.nvidia.cn/rdp/cudnn-download

​ 2.解壓下載的cudnn文件,會獲得三個文件夾:bin、include、lib;

​ 3.將這三個文件夾復制到之前CUDA的安裝路徑里,最后一個路徑名應是cuda版本號

​ 4.CUDA安裝時若是默認路徑,則一般為:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

四、pytorch安裝(清華鏡像)

1.打開   Anaconda Prompt

2.添加清華鏡像

依次輸入下列三行代碼:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

3.添加pytorch鏡像

 接着輸入下列代碼:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

4.安裝pytorch

進入官網鏈接: https://pytorch.org/

​ 1.在官網根據自己之前安裝的環境依次選擇,然后復制官網給出的命令語:

​ Install——stable——Windows——conda——python——10.2 (以cuda10.2為例)

​ 2.若按照上列方式官網提供的cuda版本過高,點擊上方的 “ install previous versions of pytroch”,選擇舊版本

​ 3.根據不同的版本搜索自己的下載語句,記住去掉 -c pytorch

conda insatll pytorch torchvision torchaudio cudatookit=10.2 (以cuda10.2為例) 

5.測試

1.win+R,輸入cmd,進入命令提示符

​2.輸入

python     # 進入>>>python環境
import torch    # 如正常則靜默
a = torch.Tensor([1.])    # 如正常則靜默
a.cuda()    # 如正常則返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn    # 如正常則靜默
cudnn.is_acceptable(a.cuda())       # 如正常則返回 "True"

6.注意事項

如果以前安裝過pytorch或者安裝失敗過,在安裝前執行下面兩條語句:

conda uninstall pytorch
conda uninstall libtorch

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM