1.anaconda下载
2.查看是否有独立显卡以确定 Pytorch的下载参数
3.安装Pytorch
1.anaconda下载
Anaconda | Individual Edition
2.查看是否有独立显卡以确定 Pytorch的下载参数
3种查看方式 :
1.任务管理器-GPU型号
2.设备管理器
3.命令行
nvidia-smi
显卡驱动网站:Official Drivers | NVIDIA
3.安装Pytorch
网站选型:
#1.先用 anaconda prompt 建一个conda environment:
conda create -n pytorch pyton=3.7 #创建
conda active pytorch #激活环境
pip list #查看安装包
#2.在这个环境中安装 Pytorch 安装参数根据前面查看的硬件配置分(有显卡,没显卡)两种安装包
https://pytorch.org/
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
安装的时候看网速,我这花了十几分钟。
安装验证:
python #进入python
import torch #进入torch
torch.cuda.is_available() #是否可以访问gpu
搞定!