Win10 下载torch以及CUDA配置

下载torch和torchvision

进入该网站，下载对应的torch和torchvision版本。

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

以该图为例，cu102代表CUDA10.2版本，cp代表python版本，win当然代表window系统了，而64则是64位操作系统的意思。

一些trick：如果太多的whl看到眼花缭乱，我们可以CTAL+F快速搜索，按回车到达指定位置，效果如下。

需要注意的是torch有分CPU版本和GPU版本，如果想用CUDA的同学一定要下载GPU版本，并且如果你安装的是CUDA10.2，则千万不能去下载前缀不是cu102的whl。同理torchvision的版本也要拿捏好。

安装torch和torchvision

以anaconda的虚拟环境为例，我们打开Prompt工作台。

activate +你的虚拟环境名称
cd 你whl下载的路径
pip install torch-1.60-cp38-cp38-win_amd64.whl(你whl的名字)
pip install torchvision....（一样是你whl的名字）

至此torch和torchvision就安装完毕了。

安装CUDA

进入官网

https://developer.nvidia.com/zh-cn/cuda-downloads

选择对应的版本

下载这个，然后再等待下载完成之后，打开exe文件。一直默认操作就可以了，由于cuda比较特殊，一般不建议修改安装路径，不过exe的存放路径倒可随意。

检查是否成功

打开电脑cmd，输入nvcc -V查看cuda是否成功，如果成功会出现以下的字符。注意大小写要区分开。

最后，我们在代码中试试，导入torch之后输入该代码。

print("successful installation!") if torch.cuda.is_available() else print("fail installation")

如果成功会返回

successful installation!