1、安裝yaml依賴：sudo apt-get install python-yaml python3-yaml

2、git clone pytorch 原始碼：

git clone https://github.com/pytorch/pytorch.git

git clone https://github.com/pytorch/vision.git

3、pytorch安裝：進入pytorch目錄

git submodule update --init --recursive

sudo python3 setup.py install

4、 pytorchvision安裝：進入vision目錄

sudo apt-get install cmake

sudo python3 setup.py install

5、python3進入驗證：

import torch

import torchvision