一、环境准备
一、前提已安裝好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的下載命令
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1
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(),退出
總結
- 上一篇: 如何关掉Microsoft Office
- 下一篇: 一、线性回归