一、配置刷机
二、安装torch、torchvison
https://download.pytorch.org/whl/torch_stable.html (这是编译好的离线安装包)
本人亲测如下:
1、下载最新版本:
2、执行安装指令:
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip3 install Cython
pip3 install numpy torch-1.10.0-cp36-cp36m-linux_aarch64.whl
3、导入torch模块 测试cuda正常
4、确认torch安装正确之后,接着开始进行torchvision的安装,链接如下:GitHub - pytorch/vision at release/0.11
安装指令如下,时间可能会有点长 大概十几分钟 做好心理准备。
$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision # see below for version of torchvision to download #可以直接从github上下载
$ cd torchvision
$ export BUILD_VERSION=0.x.0 # where 0.x.0 is the torchvision version 这我用的版本torch 1.10 对应的vision应该是0.11.1
$ python3 setup.py install --user
ok 这一切装完之后就可以开始安装 稀疏卷积库 spconv了
首先,确认一下cmake版本:
cmake --version
如果提示未安装,则进行安装,版本不低于 3.13.2,
下载链接:
安装教程参考链接: