ubuntu16.04-arm版 下载pytorch

要在华为的atlas 200dk上配置一下深度学习的推理环境。

网上编译好的包不能直接用,要重新开始编译。
下载pytorch
对系统进行更新
1.sudo apt update && sudo apt upgrade
创建python的虚拟环境,因为这个架构还不支持conda,所以要用venv创建虚拟环境。
2.python3 -m venv pytorch
如果是指定版本的,将python3改为python3.6,python3.7即可。但是要先执行如下代码。
3.sudo apt-get install python3.6-venv
接下来使用git下载源代码。
4.git clone --recursive https://github.com/pytorch/pytorch
5.cd pytorch
切换到自己想要的版本,比如V1.0.1
6.git checkout tags/v1.0.1 -b build
7.git submodule update --init --recursive
在编译pytorch之前,我们还是需要安装依赖的python包。安装依赖包:
8.pip install -r requirements.txt
9.设置环境变量

export NO_CUDA=1
export NO_DISTRIBUTED=1
export NO_MKLDNN=1
export NO_NNPACK=1
export NO_QNNPACK=1

10.对pytorch进行编译,这个要很久很久。

python setup.py build

ubuntu16.04-arm版 下载pytorch_第1张图片
11.编译完成后,利用命令进行安装。

python setup.py install

你可能感兴趣的:(ubuntu16.04,机器学习,深度学习,pytorch)