win10安装pytorch1.4.0过程记录

  1. 首先确认显卡情况
    找到显卡驱动,在系统信息中找到对应的CUDA版本

win10安装pytorch1.4.0过程记录_第1张图片

  1. 去官网下载安装CUDA和cudnn(参考博文)

CUDA:https://developer.nvidia.com/cuda-downloads
cudnn:https://developer.nvidia.com/rdp/cudnn-download

  1. 安装Python
    需要看清楚python是32位还是64位,目前大部分新版本的Pytorch都只有64位的,所以python要安装64位的。这里选的是py3.7

  2. 安装pytorch
    许多文章介绍的都是用pip和conda,然而我试过了都不行(没有1.4.0版本的),最后去官网目录挨个找对应版本的下载,目前没有cuda10.2的,最多10.1,就下101的,python版本和位数要对应(如果不对应,会无法安装成功,即使改了名字安装成功了,也会报import torch._C的错误):
    win10安装pytorch1.4.0过程记录_第2张图片
    然后直接 pip install torch……………… 即可安装成功(需要先安装numpy)。

你可能感兴趣的:(Windows,Python)