win10安装pytorch

  1. 安装cuda10(官网支持的是8.0,9.0,10.0,由于之前安装的是9.2,结果安装好后导入错误,遂安装了最新的cuda10),安装方法直接去官网下载exe文件即可。
  2. 安装cudnn7.3,去官网下载cudnn,按照官网安装方法安装(拷贝几个文件即可)。
  3. 安装pytorch,按照官网方法安装即可(选择对应的版本,pip安装)。
  4. 测试:import torch没有报错即成功;print(torch.cuda.is_available())为true即成功。

发表评论

邮箱地址不会被公开。 必填项已用*标注