如何在window上安装Pytorch?有详细的安装和验证过程。

  1. 进入anaconda prompt命令行,创建一个独立运行的环境(否则很容易安装失败),然后激活这个环境.

    conda create -n pytorch_gpu pip python=3.6

    conda activate pytorch_gpu

  2. 为了提升安装的速度,可以使用国内镜像来下载安装包。

    阿里云的源
    conda config --add channels http://mirrors.aliyun.com/pypi/simple/

  3. 进入pytorch官网,选择适合自己的版本,我选择的是Stable(1.3)、Windows、Conda、Python3.6、10.1,输入下面的命令。

    conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

在这里插入图片描述

  1. 看到下面的图说明安装完毕。

在这里插入图片描述

  1. 验证一下是否安装成功,先输入python,进入python环境,然后直接输入下面的命令。

    from future import print_function
    import torch
    x = torch.rand(5, 3)
    print(x)
    import torch
    torch.cuda.is_available()

    如果打印出下面的内容,说明安装成功。

    tensor([[0.3380, 0.3845, 0.3217],
            [0.8337, 0.9050, 0.2650],
            [0.2979, 0.7141, 0.9069],
            [0.1449, 0.1132, 0.1375],
            [0.4675, 0.3947, 0.1426]])
    True
    

    在这里插入图片描述

  2. 输入exit()退出python环境,回到了pytorch_gpu虚拟环境中。为了能在juputer notebook中使用该虚拟环境,需要安装下面的程序。

    conda install nb_conda_kernels

    打印如下内容时,说明安装完毕。

    在这里插入图片描述

  3. 输入jupyter notebook命令,在New下拉菜单中选择conda env:pytorch_gpu虚拟环境创建新文件。

    在这里插入图片描述
    输入如下命令

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

    如果打印如下信息,说明安装成功,可以通过虚拟环境使用pytorch了。

    在这里插入图片描述
    如果是已存在的文件,通过kernel下的Change kernel下选择自己要用的环境
    在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值