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.csdn.net/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那块卡,等用起来再看看吧