美文网首页pytorch学习
pytorch基础学习(一) pytorch的安装

pytorch基础学习(一) pytorch的安装

作者: SnowPye | 来源:发表于2020-05-28 12:48 被阅读0次

    1. 下载安装

    pytorch是目前应用最宽泛的机器学习库,因其代码框架明晰而广受欢迎。
    pytorch的安装也非常简单,进入pytorch官网https://pytorch.org/,可以看到下图:


    根据自己的系统版本,安装方式,cuda版本进行选择,直接安装即可。
    建议使用conda进行安装,进入anaconda的官网进行下载安装https://www.anaconda.com/

    2. 可能出现的问题

    因为pytorch的网站是在国外的,国内安装很可能因为网络问题安装不上,此时可以通过添加清华源或从pytorch库网站https://download.pytorch.org/whl/torch_stable.html下载安装包直接安装(注意先装torch,再装torchvision)。

    3.安装完成测试

    安装完成后,运行以下语句进行测试:
    CPU版:

    import torch
    import torchvision
    
    t = torch.tensor([1, 2, 3], dtype=torch.float32)
    print (type(t))  # <class 'torch.Tensor'>
    print (t.dtype)  # torch.float32
    print (t.device)  # cpu
    
    output:
    <class 'torch.Tensor'>
    torch.float32
    cpu
    

    GPU版:

    import torch
    import torchvision
    
    t = torch.tensor([1, 2, 3], dtype=torch.float32).cuda()
    print (type(t))  # <class 'torch.Tensor'>
    print (t.dtype)  # torch.float32
    print (t.device)  # cuda:0
    
    output:
    <class 'torch.Tensor'>
    torch.float32
    cuda:0
    

    相关文章

      网友评论

        本文标题:pytorch基础学习(一) pytorch的安装

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