准备
su进入root,避免遇到权限不足情况
python3 -m venv -xxx (创建的就是一个python环境和pytorch无关,可以安装任意环境的pytorch)
source进入虚拟环境
sudo apt-get install libopenblas-dev libblas-dev m4 cmake cython3 ccache
修改源码文件
cd pytorch / third_party
# 删除原有cpuinfo目录
sudo rm -r cpuinfo
# 克隆算能的cpuinfo目录
sudo git clone https://github.com/sophgo/cpuinfo.git
依赖项
apt install python3
apt install python-is-python3 python-dev-is-python3
apt autoremove
# 下载好python3后发现没有pip、yaml等工具
apt install python3-pip
pip install pyyaml
pip install typing_extensions
pip install ninja-build
python3 -m pip install numpy --upgrade -i https://pypi.tuna.tsinghua.edu.cn/simple
环境变量配置
export USE_CUDA=0 #不编译CUDA版本
export USE_DISTRIBUTED=0 #不支持分布式
export USE_MKLDNN=0 #不支持MKLDNN
export MAX_J0BS=16 #最大线程数 ,按需填写
export BUILD_TEST=0
构建指令
python setup.py bdist_wheel
构建完成后将会在dist目录下生成whl安装包
然后进入dist目录 pip install xxx.whl