pytorch安装(无GPU)
首先进入pytorch首页
链接:https://pytorch.org/get-started/locally/
输入命令:
conda install pytorch torchvision cpuonly -c pytorch
pytorch安装(有GPU)
输入命令:
conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses
检验pytorch是否安装成功(有GPU)
python
import torch
torch.cuda.is_vailable