1.anaconda
建議安裝在D盤,官網正常安裝即可,一般可以安裝windows版本
安裝成功后,可以在電腦應用里找到:
2.創建虛擬環境
打開anaconda prompt,
可以用conda env list 查看現有的環境,一般打開默認base
conda create -n 虛擬環境名 python==版本號
激活環境:conda activate 虛擬環境名
可能會提示要先初始化
那可以win+r打開cmd終端,輸入conda init
然后再回到conda終端激活虛擬環境即可
3.虛擬環境里安裝pytorch(GPU版)
(1)先判斷有沒有CUDA
打開電腦任務管理器,可以在電腦搜索框搜索
打開,頁面左側,選性能按鈕,若右上角為NVIDIA開頭,表示有CUDA,如果是Intel開頭,表示沒有,如果左側有多個GPU選項,如GPU0,GPU1,只要其中有一個為NVIDIA開頭即可
確定有CUDA,安裝GPU版本,沒有,安CPU版本,兩者對完成任務沒有影響,只是GPU訓練更快一些
(2)GPU三部分
硬件
即剛剛查看的顯卡,要看顯卡的算力,可以去網頁或者官網查看,確定算力后,查看需要的驅動器版本
驅動器
用來驅動顯卡的
查看自己電腦上驅動器的版本,終端命令:nvidia-smi,版本為12.8
如果版本過低,不足以支持算力,或者未安裝驅動,請到官網下載或者更新
官網網址:下載 NVIDIA 官方驅動 | NVIDIA
做手動驅動搜索選擇自己的型號,下載即可
CUDA Runtime
應用軟件用來使用顯卡的程序,在我們下載pytorch時需要選擇版本,原則上CUDA Runtime版本應該不高于驅動器,即CUDA Driver版本,否則可能會導致部分功能無法正常使用,如12.8的驅動器,對應的pytorch CUDA版本應該低于12.8
(3)安裝pytorch
pytorch官網,選擇自己需要的選項,然后復制下方指令,進入虛擬環境的終端進行安裝即可,正常情況下,一般使用conda包,但因為官網暫時關閉了conda包下載,所有使用pip下載,差不多
(4)檢查pytorch安裝是否成功
conda activate env_name//你創建的虛擬環境名,激活虛擬環境
conda list//查看有沒有pytorch或者torch,torchvision,torchaudio
python//輸入python
import torch//輸入,導入torch包
torch.cuda.is_available()
輸入conda list 后列表中有:
最后輸出true,表示安裝成功
4.PyCharm
官網安裝pycharm,專業版免費試用30天,社區版免費,一般用社區版就可以
下載后創建一個項目,在左上,選文件(Files)—設置(Settings)—項目(project)—python解釋器(python Interpreter)
右側,添加解釋器(Add Interpreter)—添加本地解釋器—左側選conda解釋器
瀏覽剛剛安裝了python的虛擬環境,一般是anacoda的安裝文件—env文件—虛擬環境文件—python.exe
選使用現有環境,使用創建的虛擬環境,ok
接下來就可以用pycharm運行代碼了
更詳細的操作可以看B站視頻:
最詳細的 Windows 下 PyTorch 入門深度學習環境安裝與配置 CPU GPU 版 | 土堆教程_嗶哩嗶哩_bilibili