美文网首页
window下安装pytorch

window下安装pytorch

作者: yimengtianya1 | 来源:发表于2020-09-23 17:06 被阅读0次

    1、下载pytorch安装包

    需要下载torch、torchvision两个安装包。
    1.1 打开pytorch官网https://pytorch.org/

    图片.png
    1.2 进入下载包界面
    选择pip安装模式,语言版本,由于我电脑没有Nvidia的显卡,所以CUDA选择的None。如果有显卡的,查一下显卡适配的CUDA的版本,对应下载即可。
    最后会出现安装命令:
    pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
    
    图片.png

    这里有两种安装方法:
    (1)将安装包下载好了,再在cmd命令下,采用pip命令进行安装。

    pip install [安装包文件路径] 
    

    (2)直接采用官网推荐的run this command里的命令进行安装。

    pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
    

    这里,我采用的是第一种安装方案。主要是担心,直接安装网络下载比较慢,容易中途断掉,前功尽弃。
    (3) 将图中的网址打开,可以看到对应的安装包。这里,对应安装命令里对应的版本,进行下载。


    图片.png

    我下载的版本是:

    torch-1.6.0+cpu-cp36-cp36m-win_amd64.whl
    torchvision-0.7.0+cpu-cp36-cp36m-win_amd64.whl
    

    2、安装

    打开cmd窗口,直接执行:

    pip install [安装包路径地址] 
    

    先安装torch-1.6.0,再安装torchvision-0.7.0即可。

    3、torch的windows依赖

    torch在windows运行,依赖VC_redist.x64.exe。下载链接地址如下:
    https://download.visualstudio.microsoft.com/download/pr/fd5d2eea-32b8-4814-b55e-28c83dd72d9c/952A0C6CB4A3DD14C3666EF05BB1982C5FF7F87B7103C2BA896354F00651E358/VC_redist.x64.exe
    下载安装后,重启电脑即可。

    4、验证安装

    在IDE中执行以下代码,正常执行,即表示安装成功。若报错,可能是第3步安装的windows依赖有问题,按提示安装即可。

    from __future__ import print_function
    import torch
    
    x=torch.rand(3,3)
    print(x)
    

    相关文章

      网友评论

          本文标题:window下安装pytorch

          本文链接:https://www.haomeiwen.com/subject/iuqtuktx.html