Firefly开源社区

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

---

作者: -Forest-    时间: 2020-4-20 11:09
标题: rk3399ubuntu18.04 pytorch环境搭建


整整配置了一个周末，把踩过的坑记录下来，以后参考用：
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. 
    
12. deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
    
13. deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
    
14. deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
    
15. deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
    
16. deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
    
17. deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
    
18. deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic main universe restricted
    
19. deb-src http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic main universe restricted
    
20. 
    
21. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
    
22. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
    
23. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
    
24. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
    
25. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
    
26. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
    
27. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main universe restricted
    
28. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main universe restricted
    
29. 
    
30. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial main multiverse restricted universe
    
31. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-security main multiverse restricted universe
    
32. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-updates main multiverse restricted universe
    
33. deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-backports main multiverse restricted universe
    
34. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial main multiverse restricted universe
    
35. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-security main multiverse restricted universe
    
36. deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ xenial-updates main multiverse restricted universe
    
37. 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. 
   
3. sudo apt-get upgarde#这条不用
   
4. sudo apt-get install python-pip python-dev build-essential
   
5. 
   
6. sudo apt-get install python3-pip python3-dev build-essential
   
7. 
   
8. sudo apt-get install gedit
   
9. 
   
10. sudo apt-get install openssl*
    
11. 
    
12. sudo apt-get install ipython3
    
13. 
    
14. sudo apt-get install minicom
    
15. 
    
16. sudo apt-get install cmake git
    
17. 
    
18. sudo apt-get install inetutils-ping
    
19. 
    
20. 安装中文库和输入法：可选
    
21. 
    
22. sudo apt-get install scim scim-pinyin
    
23. 
    
24. 安装火狐浏览器：可选
    
25. 
    
26. sudo apt-get install firefox
    
27. 
    
28. 安装依赖：
    
29. 
    
30. 
    
31. sudo apt-get install python3-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev
    
32. 
    
33. sudo apt-get install libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
    
34. 
    
35. 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

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