Firefly开源社区

标题: rk3399ubuntu18.04 pytorch环境搭建 [打印本页]

作者: -Forest-    时间: 2020-4-20 11:09
标题: rk3399ubuntu18.04 pytorch环境搭建
本帖最后由 -Forest- 于 2020-4-22 17:41 编辑

整整配置了一个周末,把踩过的坑记录下来,以后参考用:
1.换源:
  1. sudo vi /etc/apt/sources.lis
复制代码
备份旧源以防万一
替换为下列内容
  1. #deb http://ports.ubuntu.com/ bionic main restricted universe multiverse
  2. #deb-src http://ports.ubuntu.com/ bionic main restricted universe multiverse
  3. #deb http://ports.ubuntu.com/ bionic-updates main restricted universe multiverse
  4. #deb-src http://ports.ubuntu.com/ bionic-updates main restricted universe multiverse
  5. #deb http://ports.ubuntu.com/ bionic-security main restricted universe multiverse
  6. #deb-src http://ports.ubuntu.com/ bionic-security main restricted universe multiverse
  7. #deb http://ports.ubuntu.com/ bionic-backports main restricted universe multiverse
  8. #deb-src http://ports.ubuntu.com/ bionic-backports main restricted universe multiverse
  9. #deb http://ports.ubuntu.com/ubuntu-ports/ bionic main universe restricted
  10. #deb-src http://ports.ubuntu.com/ubuntu-ports/ bionic main universe restricted #Added by software-properties

  11. deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
  12. deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
  13. deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
  14. deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
  15. deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
  16. deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
  17. deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic main universe restricted
  18. deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic main universe restricted

  19. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
  20. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
  21. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
  22. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
  23. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
  24. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
  25. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main universe restricted
  26. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main universe restricted

  27. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial main multiverse restricted universe
  28. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-security main multiverse restricted universe
  29. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-updates main multiverse restricted universe
  30. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-backports main multiverse restricted universe
  31. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial main multiverse restricted universe
  32. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-security main multiverse restricted universe
  33. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-updates main multiverse restricted universe
  34. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-backports main multiverse restricted universe
复制代码

更新源
  1. apt-get update
复制代码

2.更改python默认版本
新系统默认2.7,更换为系统自带3.6
  1. rm /usr/bin/python
  2. ln -s /usr/bin/python3.6 /usr/bin/python
复制代码

查看是否替换成功
  1. python -V
复制代码

3.安装环境
  1. sudo apt-get update

  2. sudo apt-get upgarde#这条不用
  3. sudo apt-get install python-pip python-dev build-essential

  4. sudo apt-get install python3-pip python3-dev build-essential

  5. sudo apt-get install gedit

  6. sudo apt-get install openssl*

  7. sudo apt-get install ipython3

  8. sudo apt-get install minicom

  9. sudo apt-get install cmake git

  10. sudo apt-get install inetutils-ping

  11. 安装中文库和输入法:可选

  12. sudo apt-get install scim scim-pinyin

  13. 安装火狐浏览器:可选

  14. sudo apt-get install firefox

  15. 安装依赖:


  16. sudo apt-get install python3-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev

  17. sudo apt-get install libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev

  18. apt-get install python3-dev python3-pip
复制代码

5 安装opencv
  1. apt search opencv
复制代码
可以看到python-opencv和python3-opencv。分别对应python2和python3的版本。
安装python3的版本
  1. apt-get install python3-opencv
复制代码
这个命令自动安装numpy
如果不成功可以自己编译opencv,比较麻烦,后续上传opencv源码和编译方法:opencv源码:
提取码:92oj
编译参考:
https://blog.csdn.net/qq_25680531/article/details/80854344
5 安装PyTorch

如果不想自己折腾,直接下载编译好的包安装即可
百度网盘:
提取码:fxf8
下载好后:

  1. pip3 install ./torch-1.1.0a0+7e73783-cp36-cp36m-linux_aarch64.whl
复制代码
装好后 导入torch时会报错:
  1. File "/home/firefly/.local/lib/python3.6/site-packages/torch/__init__.py", line 79, in <module>
  2.     from torch._C import *
  3. ImportError: libopenblas.so.0: cannot open shared object file: No such file or directory
复制代码

原因是缺少库:libopenblas.so
安装参考:https://blog.csdn.net/mywmy/article/details/96993537


如果想试一下编译pytorch:
pytorch1.1.0源码下载:(如果不需要新版本,网速慢就直接下载这个)

提取码:cvew
pytorch源码编译参考树莓派上编译安装pytorch
交换空间尽量在2g以上,要不内存实在不够用,编译六核全开老报错,我用三个核编译。





然后就能愉快的跑模型了













作者: Wanggp    时间: 2020-4-22 00:57
你好,我按照你的方法安装了pytroch,但是安装完成之后出现了这个请问题,请问你有出现过么?
>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/firefly/.local/lib/python3.6/site-packages/torch/__init__.py", line 79, in <module>
    from torch._C import *
ImportError: libopenblas.so.0: cannot open shared object file: No such file or directory

作者: -Forest-    时间: 2020-4-22 09:50
本帖最后由 -Forest- 于 2020-4-22 09:52 编辑
Wanggp 发表于 2020-4-22 00:57
你好,我按照你的方法安装了pytroch,但是安装完成之后出现了这个请问题,请问你有出现过么?
>>> import  ...

重新打开一个终端试一试,或者重启一下。装一下openblas
作者: LiCheng    时间: 2020-4-22 10:00
感谢楼主分享。
作者: -Forest-    时间: 2020-4-22 10:07
LiCheng 发表于 2020-4-22 10:00
感谢楼主分享。

大佬入场
作者: 九虎    时间: 2020-4-22 15:22
楼主你好,你的帖子对我帮助很大。但是我现在需要用1.4.0版本的torch,我在官网上找了个版本torch-1.4.0-cp36-cp36m-manylinux1_x86_64.whl是不是不能使用?我要升到1.4.0的版本是不是需要自己完成编译?

作者: -Forest-    时间: 2020-4-22 17:24
九虎 发表于 2020-4-22 15:22
楼主你好,你的帖子对我帮助很大。但是我现在需要用1.4.0版本的torch,我在官网上找了个版本torch-1.4.0-cp ...

你这个wheel是x86_64端的,不是arm下的,如果非要用1.4.0的话就得自己去编译。
作者: Wanggp    时间: 2020-4-22 22:06
-Forest- 发表于 2020-4-22 09:50
重新打开一个终端试一试,或者重启一下。装一下openblas

谢谢楼主,这个问题已经解决了,请问这个roc-rk3399-pc在跑深度学习的模型的时候有没有加速电路?pytorch是不是只能用cpu计算。
作者: wx_哄哄_Xy8yy    时间: 2020-4-22 23:01
楼主,用你的方法已经成功安装上torch了,但是我torchvision怎么也装不上,能不能详细讲解下torchvision怎么安装,谢谢楼主。
作者: -Forest-    时间: 2020-4-23 21:18
wx_哄哄_Xy8yy 发表于 2020-4-22 23:01
楼主,用你的方法已经成功安装上torch了,但是我torchvision怎么也装不上,能不能详细讲解下torchvision怎 ...

我一开始装torchvision也装不上,装好pytorch后自己就装上了,试试再装一遍pytorch。实在不行自己编译一下源码。
作者: -Forest-    时间: 2020-4-23 21:25
Wanggp 发表于 2020-4-22 22:06
谢谢楼主,这个问题已经解决了,请问这个roc-rk3399-pc在跑深度学习的模型的时候有没有加速电路?pytorch ...

你是说调用gpu或是npu(用扩展版接上的那个?),用pytorch是不能了,现在瑞芯微官方主力推3399pro,集成了npu,他们用自有框架rknn可以调用npu。装了pytorch玩玩得了,加速不用想了
作者: SJMa    时间: 2021-7-30 15:37
楼主您好!想请问您,我是在Pytorch下面自己修改的网络结构,我看官网说要转换成RKNN Toolkit模型才可以在npu上运行。保存好的预训练模型是.pkl格式的,可以转换吗?转换后在Python中直接调用就可以吗?
作者: zzz1997    时间: 2023-5-9 17:09
我是ubunt20的系统然后python版本为3.8,opencv按照成功了但是pytorch一直解决不了,有大佬知道该怎么弄吗




欢迎光临 Firefly开源社区 (https://dev.t-firefly.com/) Powered by Discuz! X3.1