一、前提已安裝好anaconda
二、打開cmd命令窗口
conda activate
激活base環境
conda create -n pytorch python=3.6
創建一個名稱為pytorch的環境空間,其中使用的python版本為3.6
conda env list
查看當前都有哪些環境
conda activate pytorch
激活剛安裝的環境
pip list
顯示一下環境中都有哪些包
三、Pytorch的安裝
1,先查看自己的電腦是否有GPU
Ctrl+Shift+Esc
打開任務管理器,性能
,往下拉看下是否有GPU
2,查看GPU是否支持Coder
方法一:
Win+R
,輸入DXDIAG
,顯示
,看下有沒有加速功能
方法二:
cmd
,nvidia-smi
,看下是否有詳細信息出現,這里的Driver Version必須大于396.26,否則需要去官網升級驅動
若方法一有加速,但方法二報錯,則需要將C:\Program Files\NVIDIA Corporation\NVSMI
加入到系統變量的Path中
3,有GPU的支持Conder情況下,安裝Pytorch
控制面板找NVIDA的驅動型號,我這里是10.1.0,所以去PyTorch中下載對應的支持GPU的下載命令
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
4,沒有GPU情況下,安裝Pytorch
conda install pytorch torchvision torchaudio cpuonly
第3步和第4步都在是pytorch環境激活狀態下安裝
pip list
看下當前的包都有哪些
四、Pytorch的檢驗
python
,進入python環境
import torch
,導下包看看有沒有報錯
torch.cuda.is_available()
看下是否支持GPU
quit()
,退出