- 获取TorchVision源码。
1git clone https://github.com/pytorch/vision.git -b v0.19.0 --depth 1
- 进入vision目录，编译安装TorchVision。
1 2
cd vision python3 setup.py bdist_wheel
安装完成后将在dist目录下生成torchvision*.whl文件。
- 安装验证。
1 2
pip3 install dist/torchvision*.whl python3
输入如下命令：
1 2
import torchvision print(torchvision.__version__)
如果输出版本号，则表示安装成功。
