win10+pytorch+cuda安装
1、查看cuda版本
打开nvidia控制面板,点击系统信息-->组件-->查看NVCUDA.DLL那一项。可以看到我的cuda是10.1版本的。如果没有的话,可以先去安装cuda,这里不再介绍cuda的安装方法。
2、下载pytorch。官网链接https://pytorch.org/。根据自己的具体情况进行选择,会生成一个相应的安装命令
3、用上一步自己选择后生成的命令进行安装
4、判断是否安装成功pytorch。如果输入以下命令,无报错,且有输出,则证明pytorch安装成功。
import torch a = torch.Tensor([1.]) a.cuda()