本文介绍的是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可用。
评论区