1,安裝PyTorch
本文介紹兩種安裝方式,第一種方式是通過conda安裝pytorch抖单,第二種方式是下載docker鏡像岔激,直接通過docker使用pytorch勒极。兩種方式各有利弊,第二種方式節(jié)省時間虑鼎,但是需要對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 # 啟動一個鏡像,創(chuàng)建容器
$ 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/