美文网首页PyTorch
PyTorch - 安装使用

PyTorch - 安装使用

作者: 红薯爱帅 | 来源:发表于2017-09-14 10:20 被阅读864次

    1,安装PyTorch

    本文介绍两种安装方式,第一种方式是通过conda安装pytorch,第二种方式是下载docker镜像,直接通过docker使用pytorch。两种方式各有利弊,第二种方式节省时间,但是需要对docker有所认识,第一种方式比较花时间,建议初学者使用第一种方式。

    1.1,安装方式1

    安装conda

    • 下载conda安装文件
    wget https://repo.continuum.io/archive/Anaconda2-4.4.0-Linux-x86_64.sh
    
    • 安装
    bash Anaconda2-4.4.0-Linux-x86_64.sh
    
    • 测试安装是否OK
    source ~/.bashrc
    conda list
    

    安装pytorch

    • 安装pytorch
    conda install pytorch torchvision -c soumith
    
    • 测试pytorch
    $ python
    >> import torch
    >>> print torch.Tensor(5, 3)
    
     0.0000e+00  0.0000e+00  0.0000e+00
     0.0000e+00  1.3087e+35  4.5825e-41
     8.3065e+34  4.5825e-41  1.3631e+35
     4.5825e-41  8.3067e+34  4.5825e-41
     1.2769e+01  1.8927e+02  0.0000e+00
    [torch.FloatTensor of size 5x3]
    
    >>> print torch.rand(5, 3)
    
     0.5814  0.2764  0.1848
     0.2699  0.9572  0.7068
     0.0585  0.3109  0.5892
     0.3031  0.9863  0.2319
     0.1296  0.7464  0.6265
    [torch.FloatTensor of size 5x3]
    

    1.2,安装方式2

    通过阿里云docker镜像仓库,搜索得到pytorch的镜像,下载后run,测试确认已安装pytorch,可以正常使用。一系列命令如下。
    接下来,可以在容器中部署pytorch的程序。

    $ docker login --username=account registry.cn-beijing.aliyuncs.com                           # 登陆阿里云仓库
    $ docker pull registry.cn-hangzhou.aliyuncs.com/kaiyu/pytorch-cuda75                         # 下载一个镜像
    $ docker run -it --name docker_pytorch -h docker_pytorch -p 10022:22 4a64c8b4e7f5 bash       # 启动一个镜像,创建容器
    $ ctrl + p , ctrl + q                # 退出当前容器的bash界面,容器依然运行
    $ docker attach 78ac0e785775         # 进入容器
    $ docker start 78ac0e785775          # 启动容器
    

    2,参考资料

    anaconda下载页面
    https://www.anaconda.com/download/
    anaconda - Installing on Linux
    https://docs.anaconda.com/anaconda/install/linux.html
    安装pytorch
    http://pytorch.org/

    相关文章

      网友评论

        本文标题:PyTorch - 安装使用

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