win10 下的pytorch (GPU),CUDA 安装

1 为什么要用cuda

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

win10 下的pytorch (GPU),CUDA 安装_第1张图片
以上我觉得很好说明了关系图。(请原谅图的粗糙)

win10 下的pytorch (GPU),CUDA 安装_第2张图片
in[6] 默认使用cpu, 而In[7] 则可以将张量 t 放在gpu上,"cuda:0"中的0 是一个索引,(gpu的索引?)

2 安装

2.1查询电脑支持的cuda

win10 下的pytorch (GPU),CUDA 安装_第3张图片
可以看到电脑支持cuda10.1

2.2 cuda与cudnn的下载安装

参考: 这里.

这位老哥讲的很详细。

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

2.3 pytorch 安装

注意pytorch的安装一定要和cuda相匹配!!!
打开pytorch 官网 win10 下的pytorch (GPU),CUDA 安装_第4张图片
复制以上代码(我的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 最终结果

配置成功!
win10 下的pytorch (GPU),CUDA 安装_第5张图片

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

你可能感兴趣的:(运行环境配置,pytorch,cuda,gpu,深度学习)