Firefly开源社区

12
发表新贴
打印 上一主题 下一主题

[人工智能] rk3399ubuntu18.04 pytorch环境搭建

23

积分

0

威望

0

贡献

技术小白

积分
23

rk3399ubuntu18.04 pytorch环境搭建

发表于 2020-4-20 11:09:03      浏览:15720 | 回复:12        打印      只看该作者   [复制链接] 楼主
本帖最后由 -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以上,要不内存实在不够用,编译六核全开老报错,我用三个核编译。





然后就能愉快的跑模型了












回复

使用道具 举报

11

积分

0

威望

0

贡献

技术小白

积分
11
发表于 2020-4-22 00:57:31        只看该作者  沙发
你好,我按照你的方法安装了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
回复

使用道具 举报

23

积分

0

威望

0

贡献

技术小白

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

重新打开一个终端试一试,或者重启一下。装一下openblas
回复

使用道具 举报

39

积分

0

威望

0

贡献

技术小白

积分
39
发表于 2020-4-22 10:00:04        只看该作者  地板
感谢楼主分享。
回复

使用道具 举报

23

积分

0

威望

0

贡献

技术小白

积分
23
发表于 2020-4-22 10:07:31        只看该作者  5#
LiCheng 发表于 2020-4-22 10:00
感谢楼主分享。

大佬入场
回复

使用道具 举报

3

积分

0

威望

0

贡献

吃瓜的群众

积分
3
发表于 2020-4-22 15:22:10        只看该作者  6#
楼主你好,你的帖子对我帮助很大。但是我现在需要用1.4.0版本的torch,我在官网上找了个版本torch-1.4.0-cp36-cp36m-manylinux1_x86_64.whl是不是不能使用?我要升到1.4.0的版本是不是需要自己完成编译?
回复

使用道具 举报

23

积分

0

威望

0

贡献

技术小白

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

你这个wheel是x86_64端的,不是arm下的,如果非要用1.4.0的话就得自己去编译。
回复

使用道具 举报

11

积分

0

威望

0

贡献

技术小白

积分
11
发表于 2020-4-22 22:06:03        只看该作者  8#
-Forest- 发表于 2020-4-22 09:50
重新打开一个终端试一试,或者重启一下。装一下openblas

谢谢楼主,这个问题已经解决了,请问这个roc-rk3399-pc在跑深度学习的模型的时候有没有加速电路?pytorch是不是只能用cpu计算。
回复

使用道具 举报

4

积分

0

威望

0

贡献

吃瓜的群众

积分
4
发表于 2020-4-22 23:01:30        只看该作者  9#
楼主,用你的方法已经成功安装上torch了,但是我torchvision怎么也装不上,能不能详细讲解下torchvision怎么安装,谢谢楼主。
回复

使用道具 举报

23

积分

0

威望

0

贡献

技术小白

积分
23
发表于 2020-4-23 21:18:06        只看该作者  10#
wx_哄哄_Xy8yy 发表于 2020-4-22 23:01
楼主,用你的方法已经成功安装上torch了,但是我torchvision怎么也装不上,能不能详细讲解下torchvision怎 ...

我一开始装torchvision也装不上,装好pytorch后自己就装上了,试试再装一遍pytorch。实在不行自己编译一下源码。
回复

使用道具 举报

返回列表
12
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

友情链接 : 爱板网 电子发烧友论坛 云汉电子社区 粤ICP备14022046号-2
快速回复 返回顶部 返回列表