ubuntu16.04-arm版 下载pytorch

概述https://blog.csdn.net/weixin_46969696/article/details/106930043?utm_term=arm%E5%AE%89%E8%A3%85pytorch&utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2~all~sobaiduweb~default-0-106930043&spm=3001.4430 要在华为的atlas200dk上配置一下深度学习的

https://blog.csdn.net/weixin_46969696/article/details/106930043?utm_term=arm%E5%AE%89%E8%A3%85pytorch&utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2~all~sobaIDuweb~default-0-106930043&spm=3001.4430

 

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

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

export NO_CUDA=1export NO_distributeD=1export NO_MKLDNN=1export NO_NNPACK=1export NO_QNNPACK=1

12345

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

python setup.py build

1

11.编译完成后，利用命令进行安装。

python setup.py install

总结

以上是内存溢出为你收集整理的ubuntu16.04-arm版 下载pytorch全部内容，希望文章能够帮你解决ubuntu16.04-arm版 下载pytorch所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。