查看: 198|回复: 6

如何在JETSON上运行tensorflow 与 pytorch?

[复制链接]

23

主题

36

帖子

382

积分

超级版主

Rank: 8Rank: 8

积分
382
发表于 2020-4-26 09:49:46 | 显示全部楼层 |阅读模式
回复

使用道具 举报

0

主题

2

帖子

18

积分

办事员

Rank: 1

积分
18
发表于 2020-4-27 11:53:29 | 显示全部楼层
回复

使用道具 举报

1

主题

7

帖子

61

积分

版主

Rank: 7Rank: 7Rank: 7

积分
61
发表于 2020-4-27 19:54:34 | 显示全部楼层
本帖最后由 Geary 于 2020-4-27 20:00 编辑

这个问题tensorflow serving的docker在jetson平台不兼容导致的。您可以尝试在这个连接里面找一下您需要的docker尝试一下:https://hub.docker.com/u/helmuthva
回复

使用道具 举报

0

主题

2

帖子

18

积分

办事员

Rank: 1

积分
18
发表于 2020-5-6 11:01:24 | 显示全部楼层
Geary 发表于 2020-4-27 19:54
这个问题tensorflow serving的docker在jetson平台不兼容导致的。您可以尝试在这个连接里面找一下您需要的d ...

您好,我按照链接下载了Docker,安装过程中有报错,能否再帮忙看一下。
TIM截图20200506105600.png
微信图片_20200506105541.png
回复

使用道具 举报

8

主题

42

帖子

162

积分

科员

Rank: 2

积分
162
发表于 2020-5-8 09:58:14 | 显示全部楼层
方凯 发表于 2020-5-6 11:01
您好,我按照链接下载了Docker,安装过程中有报错,能否再帮忙看一下。

您好:

如果是安装pytorch和tensorflow,需要注意pytorch和tensorflow的版本,和Jetpack的版本匹配。
您可以在如下链接找到对应的tensorflow和pytorch的版本,然后下载来进行安装。
https://elinux.org/Jetson_Zoo#TensorFlow

这种方法是由Nvidia官方来确保的,所以最推荐。

目前您参考的
https://docs.nvidia.com/deeplear ... /index.html#install,上面是针对JP4.4版本的命令,而JP4.4把cuda升级到了10.2,所以会出现不兼容的问题。但是JP4.4目前还是DP版本(开发者预览版本),坑比较多,如果不是为了尝鲜我们建议可以等一等。等到6月初JP4.4正式版本放出以后再进行尝试。

关于https://hub.docker.com/u/helmuthva的docker
由于不是NV提供的,我看到目前支持的JP版本是4.2.1,而您使用镜像应该是4.2.2的。我不太确定是否是因为这个原因造成的问题,等我有空的时候会尝试一下。然后反馈给您。

回复

使用道具 举报

1

主题

7

帖子

61

积分

版主

Rank: 7Rank: 7Rank: 7

积分
61
发表于 2020-5-8 11:18:53 | 显示全部楼层
方凯 发表于 2020-5-6 11:01
您好,我按照链接下载了Docker,安装过程中有报错,能否再帮忙看一下。

不知道您这边启动docker的时候是不是使用了 nvidia-docker 来替代docker命令,NV提供的nvidia-docker对docker做了一些处理,将GPU资源映射到docker内了,如果直接使用docker应该是有些资源在docker中无法直接访问的。你可以试试nvidia-docker命令启动docker。
回复

使用道具 举报

8

主题

42

帖子

162

积分

科员

Rank: 2

积分
162
发表于 2020-5-8 21:30:34 | 显示全部楼层
这个命令可以很方便的在JP4.2上安装tensorflow,应该比docker要方便一些。

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.14.0+nv19.10
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 点我注册

本版积分规则


快速回复 返回顶部 返回列表