安装GPU版本的pytorch

一、在 pytorch官网Get Started界面选择合适的版本，官网会给出相应的语句

 安装之后发现结果是CPU版本，重试几次之后结果还是一样

二、解决方案

可以参考博主windSeS的方法，在命令行中指定[python|cudatoolkit|pytorch]这三个版本不同的组合

但是在确定镜像源中存在我所指定的组合版本后，下载的torch仍然是CPU版本

如果出现这种情况，可以尝试使用pip安装

在pytorch官网previous pytorch version界面下找到自己cuda版本对应的pip安装命令行

我的cuda版本为11.1，选择安装低于我目前版本的10.2

 复制其中#CUDA10.2下的安装命令

# CUDA 10.2
pip install torch==1.10.0+cu102 torchvision==0.11.0+cu102 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

 最后验证是否安装成功

>>> import torch
>>> torch.cuda.is_available()
True

安装好啦

总结：还是多尝试官网给出的方案，比较简单可行。如安装中遇到问题，可以选择安装较低版本的pytorch。