鲲鹏社区首页
中文
注册
开发者
支持
积分兑换 NEW
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
提交
在线提单
论坛求助

编译TorchVision

  1. 获取TorchVision源码。
    1
    		 
    git clone https://github.com/pytorch/vision.git -b v0.19.0 --depth 1

    TorchVision版本请参照表1的依赖关系进行拉取。

  2. 进入vision目录，编译安装TorchVision。
    1
2
    		 
    cd vision
python3 setup.py bdist_wheel

    安装完成后将在dist目录下生成torchvision*.whl文件。

  3. 安装验证。
    1
2
    		 
    pip3 install dist/torchvision*.whl
python3

    输入如下命令：

    1
2
    		 
    import torchvision
print(torchvision.__version__)

    如果输出版本号，则表示安装成功。