前言
近期需要使用pytorch，安装后发现没有检测到我的gpu，用conda list看，果然是安装了cpu版本。于是把原来的pytorch卸载掉，再装GPU版本的pytorch，没想到历程如此艰难。
准备
cuda、cudnn什么的就不说了吧，这个网上一堆教程。anaconda创建环境也不说了吧，本文主要说一下坑。
目标
我要装的pytorch：
torch==1.5.0
torchvision==0.6.0（这个可以去https://pytorch.org/get-started/previous-versions/看）
cudatoolkit==10.2
你很快能在官网上得到安装的命令行，但是！！！要不就很慢，要不就下不了对应的cuda版本。网上有人说换清华镜像站，但是我发现清华镜像站也装不了，于是有了后面的折腾。
步骤
打开https://download.pytorch.org/whl/torch_stable.html
然后确定目标，cu102/torch-1.5.0-cp37-cp37m-win_amd64.whl
这个可以自己拼接：
-
cu102表示cuda10.2，类似的，cu101表示cuda10.1
-
torch后面接1.5是torch的版本号
-
cp37-cp37m表示用的是python3.7
-
win_amd64表示用的是Windows 64bit系统
后面的就是用idm下载了，超快。不要直接在pip下载，非常慢，几十kB/s，用idm，4MB/s
另外还需要下载torchvision，cu102/torchvision-0.6.0-cp37-cp37m-win_amd64.whl
也是同样的拼接方式下载。
安装
运行pip install 安装两个wheel就行拉，依赖pip会自动安装，好像依赖包在清华镜像站都有。
1 | pip install torch-1.5.0-cp37-cp37m-win_amd64.whl |
结果
终于是true了。。。