@[TOC] aimet在ubuntu18.04上安装教程

一:安装conda虚拟环境

conda create -n name python=3.8

安装对应的python环境是必须的,如何安装conda自行查找

二:安装aimet

(一):下载aimet安装包

1:AimetCommon-torch_cpu_1.26.0-cp38-cp38-linux_x86_64.whl
2:Aimet-torch_cpu_1.26.0-cp38-cp38-linux_x86_64.whl
3:AimetTorch-torch_gpu_1.26.0-cp38-cp38-linux_x86_64.whl
链接: [https://github.com/quic/aimet/releases/tag/1.26.0]
下载以后在虚拟环境下进行安装
pip install xxx.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/
即可
此处可以测试是否安装aimet_common.在虚拟环境下进入python环境输入
import aimet_common
不报错即为正常

(二):安装配套pytorch

pip install torch==1.9.1+cpu torchvision==0.10.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu -i https://pypi.tuna.tsinghua.edu.cn/simple/

(三):安装相关依赖

pip install -r reqs_pip_common.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install -r reqs_pip_torch_common.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install cumm==0.2.8 spconv==2.1.20 -i https://pypi.tuna.tsinghua.edu.cn/simple/

文件来源 https://github.com/quic/aimet/archive/refs/tags/1.26.0.zip里面的资源包(packaging/dependencies).里面资源不用全部下载,全部下载会有很多问题,可以自行尝试

(四):报错:缺少一点东西

conda install -c conda-forge blas
pip install liblapacke -i https://pypi.tuna.tsinghua.edu.cn/simple/

(五):将资源包里面的(Aimet_torch)文件夹复制到conda环境目录下和aimet_common在同一个文件下测试import aimet_torch

若环境目录下没有Aimet_torch可以这样干

注意事项

上述为cpu版本安装,gpu版本较为麻烦一些.我后面在继续测试.测试代码,可以用第三步下载的code,里面的代码不能直接运行.需要自己稍作修改.
若安装仍然存在问题,可以参照其他人(引用)的文章最后关于aimet安装环境的配置.我是这样做就可以

引用

链接: [https://blog.csdn.net/weixin_38498942/article/details/133786318]
链接: [https://blog.csdn.net/qq_39561198/article/details/125462180]
链接: [https://blog.csdn.net/weixin_38498942/article/details/118567280]