下載pytorch
https://pytorch.org/
下拉找到找到,下圖樣式
檢視自己電腦的GPU版本
-
方法1 鍵盤按住Win + R ** ,輸入cmd**
在彈出介面輸入nvidia-smi
比如,我的GUP版本號是 12.2
-
方法2
搜尋nvidia
彈出下圖所示介面點選 幫助---> 系統資訊
在彈出介面點選元件
可到下圖所示資訊,可以看到我的GPU版本是12.2.146
回到剛剛的torch介面 https://pytorch.org/
根據查詢到的GPU版本,選擇配置, 藍色框線適合我的GPU版本的
建立虛擬環境
點選Win 找到所有應用中的Anaconda3(64-bit)
點選紅框圖示,彈出介面輸入建立環境命令
conda create -n 自己環境的名字 python=x.x
自己環境的名字 : 比如我要建立pytorch框架為主的環境使用的是 python = 3.9,那麼名字為 pytorch_py39
啟用虛擬環境
activate 自己環境的名字
安裝pytorch庫
在啟用自己的環境下,將
注意!!!複製之前一定要符合自己的GPU版本,,,這裡如果你的沒有GPU,可以選則CPU版本的
選則好配置後,系統會自動生成程式碼,也就值紅色箭頭指的地方,把它複製貼上到
(這裡要注意,一定要先啟用要安裝pytorch庫的環境,然後再貼上)
彈出提示,輸入 y,即可完成安裝,顯示“done”
測試安裝是否成功
- 啟用環境 : activate 自己環境的名字
- 進入python: python
- 輸入: import torch