1.首先需要安装一个anaconda3(可以到清华镜像中去下载,方便)
2.创建虚拟环境
为了避免与电脑中已存在的环境中发生冲突,新建一个虚拟环境,具体步骤参见https://blog.csdn.net/lyy14011305/article/details/59500819?ticket=ST-54761-uVouyW9tXaHgviR7YENY-passport.csdn.net,打开Anaconda Prompt 进行以下操作,例如:conda create -n torch python=3.6是创建一个名称为torch,python是3.6版本的虚拟环境;然后激活activate torch
3.安装Pytorch
https://pytorch.org/在官网选择与安装电脑匹配的版本,运行对应的语句
其中需要知道电脑的CUDA版本,参见https://www.cnblogs.com/yingchuan-hxf/p/7710071.html
安装torchvision pip install torchvision
4.验证是否安装成功
输入python进入python环境,然后输入
>>>import torch
没有出现错误,就安装成功了!