windows下pytorch gpu版本安装

1. 安装好anaconda
2. 查看自己电脑的cuda版本

nvidia-smi

然后根据cuda版本找合适的pytorch和torchvision版本：下面链接查询官网

Previous PyTorch Versions | PyTorch https://pytorch.org/get-started/previous-versions/

这里不一定是这个版本，只要能与cuda版本对应的都可以选，我选择了torch 1.10.1版本，然后我的python版本是3.6，windows 64位的系统

去下面链接下载相应的.whl文件

https://download.pytorch.org/whl/torch_stable.html

将该文件下载到你的user路径下面，

注意这里不是下载到你的虚拟环境路径下。

下载完毕后进入你的虚拟环境，安装

pip install torch-1.10.1+cu111-cp36-cp36m-win_amd64.whl

若出现错误：

只需要更新你的pip即可解决

命令：python -m pip install --upgrade pip

更新成功后重新安装

若报错：ERROR: Package 'torch' requires a different Python: 3.6.0 not in '>=3.6.2'

将环境中python版本替换位所需要版本即可：

pip install python==3.6.2 -n “你的虚拟环境名称”

最后再重新输入安装命令即可。

安装成功后，输入python,然后输入import torch

输出True则表示安装成功！！！

安装成功之后可以将.whl移到其他路径或者删除（如果你保存那个盘的内存不够的情况下考虑，若内存足够可忽略）

参考：https://blog.csdn.net/pxyp123/article/details/123747978