Pytorch 1.8.1 with Cuda @ Windows 10

Pytorch的官方下载链接：

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

Python 版本 + Cuda版本 + 操作系统 的组合，颇为混乱。

我的系统是Windows 10，Python用Anaconda 2020.11，对应Python 3.8.5。安装有两个思路：

1. 安装独立的cuda + pytorch
2. 使用conda把cuda和pytorch一起安装

（1）的思路其实更一般。下载了cuda_11.1.0_456.43_win10.exe，2.92G，嫌太大。考虑（2）。

（2）参考Start Locally | PyTorch，有cuda 10.2 和 cuda 11.1两个选择：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

都下载不下来，太大，于是从错误信息找到对应文件的下载链接，用Linux wget下载好：

cuda 10.2 + pytorch 1.8.1:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/cudatoolkit-10.2.89-h74a9793_1.conda
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.8.1-py3.8_cuda10.2_cudnn7_0.tar.bz2

cuda 11.1 + pytorch 1.8.1:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/win-64/cudatoolkit-11.1.1-heb2d755_7.tar.bz2
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.8.1-py3.8_cuda11.1_cudnn8_0.tar.bz2

看了一下，cuda 11.1 + pytorch 1.8.1 for cuda 11.1 大很多，选了cuda10.2 + pytorch 1.8.1

conda install cudatoolkit-10.2.89-h74a9793_1.conda pytorch-1.8.1-py3.8_cuda10.2_cudnn7_0.tar.bz2

装好了！

然后试用torch.cuda.is_available()出错！

看提示，是驱动原因。可能是Windows 10自己安装的NVIDIA驱动不完整？于是去NVIDIA网站下载了最新的驱动，安装，搞定！