windows安装pytorch

windows安装pytorch

Scroll Down

本文介绍的是win10+anaconda环境下安装pytorch的GPU版本,如果安装CPU版本,看另一篇文章:三步安装pytorch-cpu(win10下安装pytorch-cpu)

安装pytorch的GPU版本流程与CPU版本安装类似,
首先,添加清华pytorch源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

然后切换到需要安装pytorch的虚拟环境下(没有虚拟环境的话创建一个,不建议安装到root虚拟环境中),安装pytorch、torchvision、cudatoolkit(cudatoolkit根据自己电脑的cuda版本自行选择)

conda install pytorch torchvision cudatoolkit=10.0

等待安装完成。

最后验证:
切换到刚才安装pytorch的虚拟环境下:

import torch
torch.cuda.is_available()

如果执行第一行命令未报错,则表示安装成功。
如果执行第二行命令后输出True,则表示GPU可用。