下载anaconda
安装成功后在开始可以找到anaconda命令窗口
创建子包避免全局
conda create -n pytorch python=3.6
conda activate pytorch 进入创建的新包
conda deactivate 退出
安装pytorch
点击进入网址选择合适的版本安装
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
如果有对应的显卡可以加速,可以选择cuda10.2 然后在控制窗口输入最后一行的指令进行安装(我这里直接选择从官方源安装,如果失败的话可以改成清华镜像源)
接着可以输入
python
import torch
torch.cuda.is_available()
查看显卡是否可以加速