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

在这里插入图片描述
11.编译完成后,利用命令进行安装。

python setup.py install
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值