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/
网友评论