win10 下的pytorch （GPU），CUDA 安装

1 为什么要用cuda

可以将cuda理解为使用电脑硬件GPU的API，有了cuda，我们就可以使用GPU。

以上我觉得很好说明了关系图。（请原谅图的粗糙）

in[6] 默认使用cpu, 而In[7] 则可以将张量 t 放在gpu上，"cuda:0"中的0 是一个索引，（gpu的索引？）

2 安装

2.1查询电脑支持的cuda

可以看到电脑支持cuda10.1

2.2 cuda与cudnn的下载安装

参考: 这里.

这位老哥讲的很详细。

这里补充下怎么找到环境变量：
1 找：
桌面上的此电脑->点击右键找属性->高级系统设置->环境变量->系统变量->双击PATH->点击新建
2 做：
将环境变量路径加上去即可。

2.3 pytorch 安装

注意pytorch的安装一定要和cuda相匹配！！！
打开pytorch 官网
复制以上代码（我的Anaconda配了个名叫pytorch的虚拟环境）

activate pytorch
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

下载速度还蛮快。

3 最终结果

配置成功！

win +R # 输入cmd 
activate pytorch  # 我虚拟环境名是pytorch
python
import torch
print(torch.cuda.is_available())