win10安装cuda,pytorch

win10安装cuda,pytorch

1、查看本机显卡类型，以及本机对应的可用cuda类型
本文使用的是getforce mx150，可用版本为cuda 9.1

2、到cuda官网下载cuda
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork
z

这里选network吧，我选的local结果好几次都是下一半失败了。

3、安装cudann
参考https://blog.youkuaiyun.com/cmat2/article/details/80407059
4、安装pytorch
这里对应python3.6请选择pytorch0.4.0
我开始装的pytorch0.3.0弄半天下下来发现不能用
安装可参考https://zhuanlan.zhihu.com/p/26871672
使用清华镜像的方法，又快又好用
5、测试
愉快的在命令输入Python，再import torch

不过我这里用的好像是AMD64不知道是不是用的不是getforce那块卡，等用起来再看看吧