win10+pytorch+cuda安装

1、查看cuda版本 打开nvidia控制面板,点击系统信息-->组件-->查看NVCUDA.DLL那一项。可以看到我的cuda是10.1版本的。如果没有的话,可以先去安装cuda,这里不再介绍cuda的安装方法。 2、下载pytorch。官网链接https://pytorch.org/。根据自己的具体情况进行选择,会生成一个相应的安装命令 3、用上一步自己选择后生成的命令进行安装 4、判断是否安
相关文章
相关标签/搜索