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安装

相关文章:

  • 2021-08-25
  • 2021-08-25
  • 2021-05-02
  • 2021-10-05
  • 2021-07-07
  • 2021-11-24
  • 2021-10-03
  • 2021-08-30
猜你喜欢
  • 2021-04-16
  • 2021-08-09
  • 2021-07-30
  • 2021-12-23
  • 2021-07-29
  • 2021-11-17
  • 2022-12-23
相关资源
相似解决方案