天天看點

PyTorch 開發環境搭建

Anaconda下載下傳及安裝

簡介

  • Anaconda

    官網: https://www.anaconda.com/
  • Anaconda

    是一個開源的Python發行版本,包含了180+的科學包及其依賴項,可以友善擷取開發所需要的包,并且對包進行管理,其中的科學包有:

    conda

    numpy

下載下傳

安裝

  • 先解除安裝電腦上安裝的

    Python

    ,因為安裝

    Anaconda

    的時候會安裝相應的

    Python

    版本
  • 打開

    cmd

    指令行視窗檢查安裝情況,如下圖所示說明未安裝

    Python

    PyTorch 開發環境搭建
  • 點選

    Anaconda

    安裝包進行安裝,其他選項按預設選擇
  • 更改安裝路徑(注:安裝路徑中不要出現中文和空格)
    PyTorch 開發環境搭建
  • 兩個都勾選上
    • 第一個:将

      Anaconda

      自動添加環境變量,建議勾選,之後不用手動添加環境變量
    • 第二個:将

      Anaconda

      所帶的

      Python3.8

      設定為系統預設的

      Python

  • PyTorch 開發環境搭建
  • 驗證是否安裝成功
    • 在終端輸入

      Python

      ,如果安裝成功會顯示

      Python

      版本号,且版本号旁邊有

      Anaconda

      等字樣
      PyTorch 開發環境搭建
      Warning中的内容是說環境未激活,後續操作會激活
    • 也可以輸入

      conda list

      檢視安裝的包和其對應版本
      PyTorch 開發環境搭建
    • 也可以打開

      Anaconda Powershell Prompt

      ,顯示

      (base)

      說明安裝成功
      PyTorch 開發環境搭建

建立新環境

  • Anaconda Powershell Prompt

  • 文法:
conda create --name <env_name> <package_names>      
    • conda

      :用來管理版本和Python環境的工具
    • create

      :建立環境
    • --name

      :為建立的環境命名,可以簡寫為

      -n

    • <env_name>

      :建立的環境名,推薦英文且不加空格
    • <package_names>

      :安裝在環境中的包名,如果有多個包可以用空格隔開
  • 示例:
conda create --name py38 python=3.8      
    • 意思是:建立一個名為“py38”的環境,環境中安裝版本為3.8的

      python

  • 過程:
PyTorch 開發環境搭建
    • 輸入

      y

      ,安裝所需要的包
PyTorch 開發環境搭建
    • 建立完成
PyTorch 開發環境搭建

激活環境

  • 建立完成後,有兩個選項
    • To activate this environment

      :輸入指令激活環境
    • To deactivate an active environment

      :輸入指令退出環境
  • PyTorch 開發環境搭建
    當(base)變為環境名(py38)就說明環境已經被激活了
  • 激活後輸入

    pip list

    檢視環境中有哪些包
    PyTorch 開發環境搭建
    其中沒有PyTorch,接下來安裝PyTorch

PyTorch安裝

  • PyTorch

    是一個基于

    python

    的科學計算包,主要為使用者提供了兩大功能
    • 作為

      Numpy

      的替代品, 向使用者提供使用

      GPU

      的強大算力來實作神經網絡的加速
    • 為使用者提供了一個高靈活性,速度快的深度學習平台

conda換國内源

  • 為提高安裝速度可以使用國内鏡像源,提高下載下傳速度,以下操作均在終端進行
  • 檢視源
conda config --show-sources      
  • 添加源
conda config --add channels 源位址      
https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
https://mirrors.bfsu.edu.cn/anaconda/pkgs/r
https://mirrors.bfsu.edu.cn/anaconda/pkgs/msys2      
  • 删除源
conda config --remove channels 源位址      

  • 進入官網: https://pytorch.org/
  • 下拉找到對應界面
    PyTorch 開發環境搭建
  • 擷取運作代碼
    • 檢視電腦顯示卡:可以在任務管理器->性能中檢視顯示卡内容
    • 無英偉達顯示卡:

      Compute Platform

      選擇

      cpu

      ,并複制指令到

      Anaconda Powershell Prompt

      中運作
PyTorch 開發環境搭建
    • 有英偉達顯示卡:需要電腦上有單獨的英偉達的顯示卡或者英偉達的顯示卡和集顯
PyTorch 開發環境搭建
    • CUDA10.2

      ,更穩定一些
PyTorch 開發環境搭建
  • Anaconda Powershell Prompt

    ,輸入指令
PyTorch 開發環境搭建
    • y

      進行安裝
PyTorch 開發環境搭建
  • 安裝完成後輸入

    pip list

    ,檢視安裝完成的包
    PyTorch 開發環境搭建

檢驗安裝

  • Anaconda Powershell Prompt

    中輸入

    python

    ,并輸入以下代碼
import torch
torch.cuda.is_available()      
  • 如果傳回

    True

    ,說明安裝成功
    PyTorch 開發環境搭建