PyTorch

PyTorch - 安装使用

2017-09-14  本文已影响864人  红薯爱帅

1,安装PyTorch

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

1.1,安装方式1

安装conda

wget https://repo.continuum.io/archive/Anaconda2-4.4.0-Linux-x86_64.sh
bash Anaconda2-4.4.0-Linux-x86_64.sh
source ~/.bashrc
conda list

安装pytorch

conda install pytorch torchvision -c soumith
$ 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/

上一篇 下一篇

猜你喜欢

热点阅读