下载
中文
注册

AMCT(Caffe)

本章节以Ubuntu 20.04为例,详述安装前准备,版本配套信息如下:

表1 配套版本信息

类别

版本限制

获取方式

备注

操作系统及版本

EulerOS release 2.0 (SP10) aarch64

cat /etc/*release && uname -m

仅支持基于CPU量化。

操作系统及版本

Ubuntu 20.04 x86_64

请从Ubuntu官网下载对应版本软件进行安装,安装完成后查询命令为:

cat /etc/*release && uname -m

支持基于CPU,GPU的量化。

操作系统及版本

Ubuntu 20.04 aarch64

请从Ubuntu官网下载对应版本软件进行安装,安装完成后查询命令为:

cat /etc/*release && uname -m

支持基于CPU,GPU的量化。

Caffe环境

caffe-master分支

当前仅支持commit id为9b891540183ddc834a02b2bd81b31afae71b2153的版本

请参考Caffe官方指导准备Caffe环境:https://github.com/BVLC/caffe/tree/master

推荐使用源码方式安装Caffe环境,如果使用命令行方式安装,出现类似"/usr/bin/python3.7: can't open file '/usr/lib/python3.7/py_compile.py': [Error 2] No such file or directory"信息时,请参见使用命令行方式安装Caffe环境失败解决。

-

CUDA toolkit/CUDA driver

11.3或10.0

请用户自行获取相关软件包进行安装,例如可以参见如下链接获取相关toolkit包,该包中包括driver软件包。

https://developer.nvidia.com/cuda-toolkit-archive

如果使用GPU模式执行量化功能,则CUDA软件必须安装。

说明:

使用Ubuntu 20.04的情况下,推荐安装11.3版本CUDA软件。Ubuntu 20.04默认gcc版本为gcc9,如果使用10.0版本CUDA软件,因其配套gcc7版本,需要对默认的gcc版本进行降级,具体操作步骤请参见使用Ubuntu 20.04安装caffe环境时,出现cuda和gcc版本不匹配的编译报错

Python

Python3.7.x、Python3.8.x、Python3.9.x

推荐使用Python3.9.2

Ubuntu操作系统请参见安装Python3.9.2(Ubuntu)

EulerOS操作系统请参见安装Python3.9.2(EulerOS )

  • 安装依赖时,请确保服务器能够连接网络。
  • 本手册以Python3.9.2为例进行介绍,相应环境变量和安装命令以实际安装Python版本为准。

numpy

1.20.0+

请参见安装依赖

protobuf

3.13.0+