一、在 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。