配置pytorch

一只小小小飞猪發表於2024-05-04

下載pytorch

https://pytorch.org/

image-20240503170241005

下拉找到找到,下圖樣式

image-20240503170307622

檢視自己電腦的GPU版本

  • 方法1 鍵盤按住Win + R ** ,輸入cmd**

    image-20240503170930269

    在彈出介面輸入nvidia-smi

    image-20240503171154748

    比如,我的GUP版本號是 12.2

  • 方法2

    搜尋nvidia

image-20240503171437855

彈出下圖所示介面點選 幫助---> 系統資訊

image-20240503171736282

在彈出介面點選元件

可到下圖所示資訊,可以看到我的GPU版本是12.2.146

image-20240503182703261

回到剛剛的torch介面 https://pytorch.org/

根據查詢到的GPU版本,選擇配置, 藍色框線適合我的GPU版本的

image-20240503172000329

建立虛擬環境

點選Win 找到所有應用中的Anaconda3(64-bit)

image-20240503173000712

點選紅框圖示,彈出介面輸入建立環境命令

conda create -n 自己環境的名字 python=x.x

image-20240503173202331

自己環境的名字 : 比如我要建立pytorch框架為主的環境使用的是 python = 3.9,那麼名字為 pytorch_py39

啟用虛擬環境

activate 自己環境的名字

安裝pytorch庫

image-20240503180606972

在啟用自己的環境下,將

image-20240503180819193

注意!!!複製之前一定要符合自己的GPU版本,,,這裡如果你的沒有GPU,可以選則CPU版本的

選則好配置後,系統會自動生成程式碼,也就值紅色箭頭指的地方,把它複製貼上到

image-20240503181332928

(這裡要注意,一定要先啟用要安裝pytorch庫的環境,然後再貼上)

彈出提示,輸入 y,即可完成安裝,顯示“done”

測試安裝是否成功

  • 啟用環境 : activate 自己環境的名字
  • 進入python: python
  • 輸入: import torch

image-20240503182022301

相關文章