win10+pytorch+cuda安装

1、查看cuda版本

打开nvidia控制面板,点击系统信息-->组件-->查看NVCUDA.DLL那一项。可以看到我的cuda是10.1版本的。如果没有的话,可以先去安装cuda,这里不再介绍cuda的安装方法。

win10+pytorch+cuda安装

2、下载pytorch。官网链接https://pytorch.org/。根据自己的具体情况进行选择,会生成一个相应的安装命令

win10+pytorch+cuda安装

3、用上一步自己选择后生成的命令进行安装

win10+pytorch+cuda安装

4、判断是否安装成功pytorch。如果输入以下命令,无报错,且有输出,则证明pytorch安装成功。

import torch   
a = torch.Tensor([1.])   
a.cuda() 

win10+pytorch+cuda安装