PyTorch - 安装使用

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/

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容